1. 下载cuda,链接如下:不同版本的链接如下
https://developer.nvidia.com/cuda-toolkit-archive
2. 按照安装命令进行安装即可,安装完成后终端执行
cd /usr/local
ls
可以看到local下面多了个cuda9.0的文件夹(我这里安装了两个cuda版本。。)
其中cuda是链接。
3. cuda环境设置
首先终端执行
sudo gedit ~/.bashrc
在文件末尾添加如下内容(如果之前安装cuda有就不需要这一步了,不过一般之前安装过的话都不需要了)
export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
export CUDA_HOME=/usr/local/cuda
注意看上面的路径都是指向/usr/local/cuda 软连接,并没有写死指向某一个cuda版本,这也方面了我们后面做cuda版本切换,只需要切换cuda软链接指向就可以了。
4. 版本切换
终端执行
cd /usr/local
然后执行
sudo rm -rf cuda
这一步是删除原有的软链接。
最后执行
sudo ln -s /usr/local/cuda-9.0 /usr/local/cuda
这一步是创建了指向cuda-9.0的软链接。
我们可以通过终端执行
重启终端
nvcc -V
查看当前使用的cuda版本,看是否设置成功。
我设置ok后可是依然报了
libcudart.so.9.0: cannot open shared object file: No such file or directory
这个错误,
最后通过终端执行
sudo ldconfig /usr/local/cuda-9.0/lib64