Ubuntu 不同版本 cuda 和 cudnn 切换
可能不同的软件对与 cuda 和 cudnn 的版本有不同的要求,所以需要根据具体的要求来切换其版本。
ubuntu 是支持安装多个版本的 cuda 的,需要使用某个的版本时切换就可以了。对于 cudnn 则是直接重装,所谓的重装其实就是 copy 后创建个软链接就可以了。
安装 cuda 比较容易,直接运行下载的 sh 文件就可以了。
sudo sh cuda_9.0.176_384.81_linux.run
1. 切换 cuda
首先应该确保环境变量的设置不是 hard code,而是指向软链接,这样就可以直接通过修改软链接来修改 cuda 的版本。当然 hard code 的方式可以通过直接修改环境变量来修改 cuda 的版本,也是可以的。
通过如下命令进入环境变量文件
sudo gedit ~/.bashrc
检查关于 cuda 的环境变量设置,如下的设置是 hard code 的,直接指定了 cuda 的版本。
# for cuda9.0
export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
而下面的设置是指向 /usr/local/cuda 的软链接。
export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}
export LD