CUDA 9.0在Ubuntu上的安装

版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://blog.csdn.net/breeze5428/article/details/78256915

1.下载CUDA

在以下网址下载9.0的deb文件(注意选取合适的平台):

https://developer.nvidia.com/cuda-downloads

(如果当前页面为9.0以上版本,可在必应上用“cuda 9.0 download ”作为关键词检索定位相关下载网页)

2.终端安装deb文件

在终端执行如下命令:

 

sudo dpkg -i cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64.deb
sudo apt-key add /var/cuda-repo-<version>/7fa2af80.pub  #执行完上一句指令后,末尾会有此指令的提示
sudo apt-get update
sudo apt-get install cuda

3.修改环境变量

在终端打开并修改.bashrc文件

 

sudo gedit ~/.bashrc

将如下内容添加到.bashrc文件末尾:
export CUDA_HOME=/usr/local/cuda-9.0 
export LD_LIBRARY_PATH=${CUDA_HOME}/lib64 
export  PATH=${CUDA_HOME}/bin:${PATH} 

 

修改完成后,在终端输入指令,使环境变量立即生效:

 

sudo ldconfig 

4.验证

在终端输入命令,实时查看GPU的使用情况:

 

watch -n 1 nvidia-smi

 

5.CUDA 9.0和CUDA 8.0共存

 

如果已经安装CUDA 9.0,还需要安装CUDA 8.0,并使默认版本为8.0,则下载deb安装并执行如下类似指令:

 

sudo dpkg -i cuda8.deb
sudo apt-get update
sudo apt-get install cuda-8-0

并修改.bashrc文件内容。

CUDA 8有个补丁文件,最好安装上:

 

sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-cublas-performance-update_8.0.61-1_amd64.deb
sudo apt-get update  
sudo apt-get upgrade cuda-8-0

 

 

 

 

 

 

 

展开阅读全文
博主设置当前文章不允许评论。

没有更多推荐了,返回首页