CUDA 9.0在Ubuntu上的安装

1.下载CUDA

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

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

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



猜你喜欢

转载自blog.csdn.net/breeze5428/article/details/78256915