同一臺機子可以兼容多個版本的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版本。
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了。