同一台机子可以兼容多个版本的CUDA的,这两天就要这么来回折腾,稍显麻烦。
1. 查看当前机子上CUDA版本;
- nvcc --version
![](https://i-blog.csdnimg.cn/blog_migrate/ffe83562471de6a3120c24b5fb0a3618.png)
由此可以看到当前是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