同一台机子可以兼容多个版本的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版本。