ubuntu安装多版本的cuda
Posted zxj9487
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了ubuntu安装多版本的cuda相关的知识,希望对你有一定的参考价值。
1. 下载cuda
各个版本的CUDA :https://developer.nvidia.com/cuda-toolkit-archive
2. 下载完cuda后,使用命令解压(eg:笔者这里是9.0的cuda)
sudo sh cuda_9.0.176_384.81_linux.run
3. 安装过程
1 #..执行完上述命令后,是一堆协议说明
2 #直接按q退出协议说明.
3 xiaojie@xiaojie: accept/decline/quit: accept #接受协议
4
5 Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 375.26?
6 y)es/(n)o/(q)uit: n #是否显卡驱动包,由于已经安装显卡驱动,选择n
7
8 Install the CUDA 9.0 Toolkit?
9 (y)es/(n)o/(q)uit: y #是否安装工具包,选择y
10
11 Enter Toolkit Location
12 [ default is /usr/local/cuda-9.0 ]: #工具包安装地址,默认回车即可
13
14 Do you want to install a symbolic link at /usr/local/cuda?
15 (y)es/(n)o/(q)uit: n #添加链接**注意这个连接,如果你之前安装过另一个版本的cuda,
16 #除非你确定想要用这个新版本的cuda,否则这里就建议选no,因为指定该链接后会将cuda指向这个新的版本**
17
18 Install the CUDA 9.0 Samples?
19 (y)es/(n)o/(q)uit: y #安装样例
20
21 Enter CUDA Samples Location
22 [ default is /root ]: #样例安装地址默认即可
4. 执行命令
sudo gedit ~/.bashrc 打开.bashrc文件
5. 在.bashrc文件内部 添加如下指令:
如果你之前已经有了cuda版本,如下:
1 export PATH=$PATH:/usr/local/cuda-8.0/bin
2 export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-8.0/lib64/
3 export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda-8.0/lib64
只需将cuda-8.0换成cuda 如下:
1 export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
2 export PATH=$PATH:/usr/local/cuda/bin
3 export CUDA_HOME=$CUDA_HOME:/usr/local/cuda
一定要记得保存文件,一定要记得保存文件,一定要记得保存文件。
6. 执行命令,激活环境
1 source ~/.bashrc
7. cuda版本切换,删除原来的软连接
1 sudo rm -rf /usr/local/cuda
8. 建立新的软连接
1 sudo ln -s /usr/local/cuda-9.0 /usr/local/cuda
9. 验证切换是否成功
多个cuda版本,可以在/usr/local/
目录下查看自己安装的cuda版本,命令:
1 cd /usr/local/
同时,在该路径下执行如下命令
1 stat cuda
2 nvcc --version
3 cat /usr/local/cuda/version.txt
10. 如果执行完 第2 、3条命令后,显示的都是同一版本的cuda,那么则成功,否则,不一致则跳转11
11. 解决方案
执行如下命令:
1 sudo gedit ~/.bashrc
打开.bashrc文件 ,把之前步骤5的三行指令换成 如下指令:
1 export PATH=/usr/local/cuda/bin:$PATH
2 export LD_LIBRARY_PATH=/usr/local/cuda/lib64$LD_LIBRARY_PATH
3 export CUDA_HOME=/usr/local/cuda
一定要记得保存文件,一定要记得保存文件,一定要记得保存文件。
执行如下激活指令:
1 source ~/.bashrc
12. 补充安装cudnn
1 $ tar -zxvf cudnn-9.2-linux-x64-v7.1.tgz
2 $ sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
3 $ sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
4 $ sudo chmod a+r /usr/local/cuda/include/cudnn.h
5 $ sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
以上是关于ubuntu安装多版本的cuda的主要内容,如果未能解决你的问题,请参考以下文章