本文参考:
https://blog.youkuaiyun.com/abc869788668/article/details/78331047
1、检查自己的GPU是否是CUDA-capable
lspci | grep -i nvidia
2.下载安装包
(1)下载NVIDIA-Linux-x86_64-384.90.run
网址http://www.geforce.cn/drivers/results/123918
(2)下载cuda_9.0.103_384.59_linux.run,注意要下载runfile文件
网址https://developer.nvidia.com/cuda-downloads
(3)下载cudnnv7.1,我下载地时候只有一个版本支持cuda9.0
网站是https://developer.nvidia.com/rdp/cudnn-download
3.安装nvidia驱动
-
参考博客:https://blog.youkuaiyun.com/abc869788668/article/details/78331047
由于我安装时已有驱动,所以这一步跳过了 -
检查 NVIDIA Driver是否安装成功
nvidia-smi
3、安装CUDA9.0
找到下载文件的路径打开终端,执行:(root路径下不用加sudo)
sudo sh cuda_9.0.176_384.81_linux.run
显示More,按q,输入accept
一路往下“y",直到提示“是否为NVIDIA安装驱动nvidia-384?”,选择"n",因为已经安装好驱动程序了,默认是安装在/usr/local/cuda-9.0文件夹下
上图显示已经安装好,但是警告不完整安装,需执行
cuda_9.0.176_384.81_linux.run -silent -driver
配置环境变量,运行如下命令打开profile文件
sudo gedit /etc/profile
打开文件后在文件末尾添加路径,也就是安装目录,命令如下:
export PATH=/usr/local/cuda-9.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64$LD_LIBRARY_PATH