同一台机子可以兼容多个版本的CUDA的,这两天就要这么来回折腾,稍显麻烦。
1. 查看当前机子上CUDA版本;
nvcc --version
由此可以看到当前是CUDA 8.0。
2. 修改.bashrc文件。
export PATH=$PATH:/usr/local/cuda-8.0/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-8.0/lib64
export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda-8.0/lib64
把以上这种添加变量的方式改为如下:
export PATH=$PATH:/usr/local/cuda/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda/lib64
修改之后,需要刷新一下:
source .bashrc
那么下面切换为CUDA 9.0版本。
2. 从官网下载
CUDA 9.0的run文件,安装过程中,选择不建立软连接。
3. 切换CUDA 9.0,使用下面命令。
sudo rm -rf /usr/local/cuda
sudo ln -s /usr/local/cuda-9.0 /usr/local/cuda
nvcc --version
这时就可以看到当前的CUDA版本变为9.0了。