Linux普通用户安装指定cuda版本，超简单方法

假如你想安装tinny-cuda-nn，经常在cmake编译的时候，报本机的cuda版本和conda虚拟环境下安装torch版本对应的cuda不匹配。

以下是解决办法：

第一步：把 .bashrc 下的cuda指定路径给注释掉。

第二步：安装你torch版本对应的cuda版本，比如我虚拟环境安装的torch版本为：

pip install torch==2.0.0 torchvision==0.15.1 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu118

再安装torch指定的cuda版本，比如cuda-11.8

conda install -c "nvidia/label/cuda-11.8.0" cuda-toolkit

其他版本可以去这里查Cuda :: Anaconda.org

第三步：查看你所在的虚拟环境ncvv -V 的输出是否为 cuda11.8

第四步：去编译

tiny-cuda-nn$ cd bindings/torch
tiny-cuda-nn/bindings/torch$ python setup.py install

显示安装成功了！

 

第五步：把第一步的注释给去掉，恢复原来的cuda路径。