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

问题

在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,安装成功。