pip安装tiny-cuda-nn编译时出现-lcuda找不到情况

最新推荐文章于 2026-05-02 09:56:33 发布

原创最新推荐文章于 2026-05-02 09:56:33 发布 · 3.2k 阅读

12 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#pip #linux #python

收录于

在tiny-cuda-nn/binds/torch目录下使用python setup.py install时遇到找不到-lcuda的问题。通过检查发现cuda库位于/usr/local/cuda/lib64，但实际需要的是libcudart.so。修改setup.py文件中的库配置后，问题得以解决。

问题

在tiny-cuda-nn/binds/torch下安装python setup.py install出现如下错误：

/home/user/miniconda3/compiler_compat/ld: cannot find -lcuda
collect2: error: ld returned 1 exit status
error: command '/usr/bin/g++' failed with exit code 1

问题分析

g++ -l 指令用于加载库
而cannot find -lcuda说明找不到cuda这个库
从-L/usr/local/cuda/lib64可以看出，cuda这个库应该在/usr/local/cuda/lib64这个位置。
进入/usr/local/cuda/lib64，查看有没有这个库。

/usr/local/cuda/lib64$ ls | grep cuda
libcudadevrt.a
libcudart.so
libcudart.so.11.0
libcudart.so.11.7.99
libcudart_static.a

实际上libcudart.so就是我们需要的那个。

解决方法

修改setup.py中L166行libraries=["cuda", "cudadevrt", "cudart_static"],，改为libraries=["cudart", "cudadevrt", "cudart_static"],
重新运行python setup.py install,安装成功。