码迷,mamicode.com
首页 > 系统相关 > 详细

ubuntu安装多版本的cuda

时间:2019-08-19 22:46:25      阅读:129      评论:0      收藏:0      [点我收藏+]

标签:enter   ln -s   协议   lib64   target   rar   工具   使用   退出   

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

标签:enter   ln -s   协议   lib64   target   rar   工具   使用   退出   

原文地址:https://www.cnblogs.com/zxj9487/p/11380079.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!