Ubuntu16.04下为Titan RTX 安装显卡驱动、CUDA、cudnn

1.禁用自带的 nouveau nvidia驱动

sudo vim /etc/modprobe.d/blacklist.conf
在文件夹最末尾处添加命令

blacklist nouveau
options nouveau modeset=0

2.更新内核

sudo update-initramfs -u

3.重启系统

sudo reboot

4.在登录界面按Ctrl + Alt + F1键(如果是ssh直接查看下面命令输出）

lsmod | grep nouveau

查看nouveau模块是否被加载,如果什么都没有输出,则执行下一步，不行的话重新操作1-3

5.下载显卡驱动

在英伟达驱动下载页面上选择对应的系统和显卡型号，下载驱动
https://www.nvidia.com/Download/index.aspx

Titan RTX的440.44驱动链接
http://us.download.nvidia.com/XFree86/Linux-x86_64/440.44/NVIDIA-Linux-x86_64-440.44.run
直接下载到服务器上

wget http://us.download.nvidia.com/XFree86/Linux-x86_64/440.44/NVIDIA-Linux-x86_64-440.44.run

cd到你下载的显卡驱动的路径

sudo chmod 777 NVIDIA-Linux-x86_64-440.44.run(驱动文件名)  #对驱动文件赋权
sudo service lightdm stop  #关闭集显

6.安装驱动

sudo bash NVIDIA-Linux-x86_64-440.44.run

按照提示安装

安装完成后可以使用nvidia-smi命令测试一下

出现上面的显卡信息就说明驱动已经安装成功了

7.重启

重启一下，准备安装cuda和cudnn

安装cuda-10.0

从官网上下载cuda-10.0的.run文件
CUDA Toolkit

安装

sudo bash cuda_10.0.130_410.48_linux.run

Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 410.48?
(y)es/(n)o/(q)uit: no       不安装cuda带的驱动Install the CUDA 10.0 Toolkit?
(y)es/(n)o/(q)uit: yesEnter Toolkit Location[ default is /usr/local/cuda-10.0 ]: Do you want to install a symbolic link at /usr/local/cuda?
(y)es/(n)o/(q)uit: yInstall the CUDA 10.0 Samples?
(y)es/(n)o/(q)uit: yEnter CUDA Samples Location[ default is /home/user ]:

设置环境变量

针对当前用户
打开主目录下的 .bashrc文件添加如下路径，vim ~/.bashrc

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.0/lib64
export PATH=$PATH:/usr/local/cuda-10.0/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-10.0

运行source ~/.bashrc

针对所有用户
打开/etc/profile文件添加如下路径 sudo vim /etc/profile

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.0/lib64
export PATH=$PATH:/usr/local/cuda-10.0/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-10.0

验证cuda安装

nvcc -V

有上面这样的输出说明安装成功

安装CUDNN

在nvidia官网上下载cudnn的deb包
CUDNN-link

下载
cuDNN Runtime Library for Ubuntu16.04 (Deb)
cuDNN Developer Library for Ubuntu16.04 (Deb)
cuDNN Code Samples and User Guide for Ubuntu16.04 (Deb)
这三个文件
依次安装

sudo dpkg -i libcudnn7_7.6.2.24-1+cuda10.0_amd64.deb
sudo dpkg -i libcudnn7-dev_7.6.2.24-1+cuda10.0_amd64.deb
sudo dpkg -i libcudnn7-doc_7.6.2.24-1+cuda10.0_amd64.deb

验证cudnn安装
上面安装的第三个包中有官方给的示例，可以用来验证cudnn是否安装成功，示例代码保存在/usr/src/cudnn_samples_v7/中

cp -r /usr/src/cudnn_samples_v7 ~/cudnn-sample
cd ~/cudnn-sample/mnistCUDNN/
make clean && make -j16
./mnistCUDNN

如果运行./mnistCUDNN最后输出test passed，就代表cudnn安装成功