先介绍下我自己配置的环境

  • Ubuntu 16.04
  • GTX2080ti显卡
  • NVIDIA 418.3
  • CUDA 10.0
  • cuDNN 7.6.2

以下教程针对从零开始的用户,若系统中已装有CUDA和cuDNN,却在当前用户下无法使用,请添加环境变量

安装NVIDIA显卡驱动

1.先在NVIDIA官网上下载对应的驱动程序,可根据自己的GPU的型号下载相应的.run文件
例如NVIDIA-Linux-x86_64-3xx.xx.run形式的文件名

2.禁用开源nouveau驱动(非常重要)
Ubuntu系统集成的显卡驱动程序是nouveau,它是第三方为NVIDIA开发的开源驱动,我们需要先将其屏蔽才能安装NVIDIA官方驱动。
将驱动添加到黑名单blacklist.conf中,但是由于该文件的属性不允许修改。所以需要先修改文件属性。
查看属性

1
sudo ls -lh /etc/modprobe.d/blacklist.conf

修改属性
1
sudo chmod 666 /etc/modprobe.d/blacklist.conf

用gedit编辑器打开
1
sudo gedit /etc/modprobe.d/blacklist.conf

在该文件后添加一下几行:
1
2
3
4
5
blacklist vga16fb
blacklist nouveau
blacklist rivafb
blacklist rivatv
blacklist nvidiafb

3.开始安装
先按Ctrl + Alt + F1到控制台,关闭当前图形环境
1
sudo service lightdm stop

再安装驱动程序
1
2
sudo chmod a+x NVIDIA-Linux-x86_64-xxx.run
sudo ./NVIDIA-Linux-x86_64-xxx.run -no-x-check -no-nouveau-check -no-opengl-files

最后重新启动图形环境
1
sudo service lightdm start

在终端里输入:nvidia-smi ,输出以下图片的代码则安装成功

cuda 10.0安装

官方下载
网盘下载 提取码: aarn

1
2
sudo chmod a+x cuda_10.0.130_410.48_linux.run // 获取权限
sudo sh cuda_10.0.130_410.48_linux.run --tmpdir=/home/max/temp

这里加 —tmpdir 主要是直接运行后,会提示空间不足的问题
接下来进入英文选择界面按住空格键可以快速浏览
在安装过程中选项选择:
accept #同意安装
n #不安装Driver,因为已安装驱动(这里需要强调一下)
y #安装CUDA Toolkit

安装到默认目录

y #创建安装目录的软链接
n #不复制Samples,因为在安装目录下有/samples

添加环境变量

home文件下 ctrl+H显示隐藏文件 打开 .bashrc文件在最后添加

1
2
export PATH=/usr/local/cuda-10.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

终端运行如下命令,保存操作
1
source ~/.bashrc

检查cuda是否安装成功
1
2
nvcc -V
nvcc --version

cuDnn 10.0安装

官方下载
网盘下载
注意:需跟CUDA版本对应
切换到下载目录进行解压:

1
sudo tar -zxvf ./cudnn-10.0-linux-x64-xxx.tgz 

解压下载的文件,可以看到cuda文件夹,在当前目录打开终端,执行如下命令:
1
2
3
4
5
6
7
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*

在终端输入
1
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

如果出现下图所示版本信息,说明安装成功