ubuntu多cuda兼容时切换cuda版本的方法

380 阅读1分钟

切换cuda版本

查看cuda版本

方法1

cat /usr/local/cuda/version.txt

方法2

$ cd /usr/local

$ stat cuda 查看指向的哪个版本

$ ll /usr/local 查看安装的版本

cuda是一个软链接,它指向我们指定的cuda版本(注意上面在设置环境变量时,使用的是cuda,而不是cuda-9.0和cuda-10.0,这主要是为了方便我们切换cuda版本,可以让我们不用每次都去该环境变量的值)

切换cuda版本

cd /usr/local

sudo rm -rf cuda

sudo ln -s /usr/local/cuda-10.2 /usr/local/cuda

其中 9.0改为相应的自己的cuda版本

以及要改bashrc文件