实验室服务器用的版本和我的复现的cuda有部分冲突。(虽说向下兼容,但是要该的python版本,那么代码里面有的也要动太费劲了)。怎么下载的,我就不说了,这里只告诉你怎么切换
第一步:(下载好不同版本)
第二步:(查看当前编译器的CUDA 版本)其中V 是大写的
nvcc -V
这里说一下,nvcc -V和nvidia -smi是不一样的,你可以理解成前者是虚拟的,后者是实际的。
第三步:在终端输入以下命令
vi ~/.bashrc
打开后是这样,按下键盘 i 进入插入模式;插入以下代码。(看看自己的版本下载路径)
export PATH=/usr/local/cuda-11.1/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.1/lib64:$LD_LIBRARY_PATH
再按下 ESC退出插入模式。
再输入,一定记住有 : 这个冒号
退出后,然后再输入
source ~/.bashrc
使得你刚才的修改起作用。
再输入nvcc -V 看看是否修改成功。
nvcc -V