ubuntu安装并管理多版本的cuda

1. 下载cuda

各个版本的CUDA :https://developer.nvidia.com/cuda-toolkit-archive

2. 下载完cuda后,使用命令解压

sudo sh cuda_9.0.176_384.81_linux.run

3. 安装过程
执行完上述命令后,是一堆协议说明
直接按q退出协议说明.
然后需要输入

accept/decline/quit: accept  #接受协议
 Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 375.26? 
  y)es/(n)o/(q)uit: n  #是否显卡驱动包,由于已经安装显卡驱动,选择n
  
  Install the CUDA 9.0 Toolkit?
  (y)es/(n)o/(q)uit: y #是否安装工具包,选择y
 
 Enter Toolkit Location
 [ default is /usr/local/cuda-9.0 ]: #工具包安装地址,默认回车即可
 
 Do you want to install a symbolic link at /usr/local/cuda?
 (y)es/(n)o/(q)uit: n #添加链接**注意这个连接,如果你之前安装过另一个版本的cuda,
              #除非你确定想要用这个新版本的cuda,否则这里就建议选no,因为指定该链接后会将cuda指向这个新的版本**
 
 Install the CUDA 9.0 Samples?
 (y)es/(n)o/(q)uit: y #安装样例
 
 Enter CUDA Samples Location
  [ default is /root ]:  #样例安装地址默认即可

4. 执行命令

sudo gedit ~/.bashrc 打开.bashrc文件

5. 在.bashrc文件内部 添加如下指令:

如果你之前已经有了cuda版本,如下:

 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

只需将cuda-8.0换成cuda 如下:

 export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
 export PATH=$PATH:/usr/local/cuda/bin
 export CUDA_HOME=$CUDA_HOME:/usr/local/cuda

一定要记得保存文件,一定要记得保存文件,一定要记得保存文件。

6. 执行命令,激活环境

1 source ~/.bashrc

7. cuda版本切换,删除原来的软连接

sudo rm -rf /usr/local/cuda

8. 建立新的软连接

 sudo ln -s /usr/local/cuda-9.0 /usr/local/cuda

9. 验证切换是否成功

多个cuda版本,可以在/usr/local/目录下查看自己安装的cuda版本,命令:

 cd /usr/local/

同时,在该路径下执行如下命令

 stat cuda
 nvcc --version
 cat /usr/local/cuda/version.txt

10. 如果执行完 第2 、3条命令后,显示的都是同一版本的cuda,那么则成功,否则,不一致则跳转11

11. 解决方案

执行如下命令:

sudo gedit ~/.bashrc 

打开.bashrc文件 ,把之前步骤5的三行指令换成 如下指令:

 export PATH=/usr/local/cuda/bin:$PATH
 export LD_LIBRARY_PATH=/usr/local/cuda/lib64$LD_LIBRARY_PATH
 export CUDA_HOME=/usr/local/cuda

一定要记得保存文件,一定要记得保存文件,一定要记得保存文件。

执行如下激活指令:

source ~/.bashrc

12. 补充安装cudnn

tar -zxvf cudnn-9.2-linux-x64-v7.1.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

发布了12 篇原创文章 · 获赞 3 · 访问量 2915

猜你喜欢

转载自blog.csdn.net/weixin_39833897/article/details/104560401