同一台机子可以兼容多个版本的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。主要内容包括查看当前CUDA版本的方法、修改.bashrc文件来配置环境变量的过程,以及通过软链接方式快速切换到不同版本的具体步骤。
6551





