Ubuntu18.04 安装指定版本的 NVIDIA驱动、CUDA以及CUDNN

Ubuntu18.04下安装指定版本的NVIDIA驱动、CUDA以及CUDNN

写在前面

在开始之前，需要安装一个全新的ubuntu 18.04，我安装的是系统是：

本文从零开始，介绍指定软件的安装步骤，内容包括：

• 安装 gcc 和其他依赖
• 卸载原驱动 nouveau
• 安装指定版本nvidia驱动
• 安装指定版本的cuda
• 安装指定版本的cudnn

在安装之前需要把指定版本的待安装的软件从官网上下载下来。下载的文件最后放在英文目录下，比如我将下载的文件都放在了文件夹 /home/xu/download/tftools 下

本文介绍的各个待安装文件的版本如下：

• ubuntu版本：18.04
  
• Nvidia驱动：410.79
  
• cuda：10.0.x
  
• cudnn：7.4.x
  

安装gcc以及其他依赖包

nvidia驱动的安装依赖gcc编译器，因此需要安装gcc。同时，一并安装了其他需要的依赖包。
安装命令：

sudo apt-get install -y build-essential cmake git unzip zip python-dev python3-dev python-pip python3-pip

查看安装版本：

gcc -v

卸载原驱动 nouveau

新安装的ubuntu18.04系统自带了显卡驱动 nouveau（可以通过命令 lsmod | grep nouveau 查看，如果有输出则证明驱动存在），因此，在安装新的驱动之前，需要将其卸载/禁用。
步骤如下：

• 在 /etc/modprobe.d/blacklist.conf 文件中添加以下两行：

blacklist nouveau
options nouveau modeset=0

• 执行命令

sudo update-initramfs -u

• 重启

reboot

• 验证，没有任何输出则证明成功卸载驱动

lsmod | grep nouveau

安装Nvidia驱动

步骤如下：

• 进入文件夹目录

cd /home/xu/download/tftools

• 给文件加上可执行权限

sudo chmod u+x NVIDIA-Linux-x86_64-410.79.run

• 运行安装文件

sudo ./NVIDIA-Linux-x86_64-410.79.run

注意，运行安装文件的时候，需要做几个选择：
- continue installation
- 32-bit的兼容问题，选择 ok
- libglvnd，选择安装和覆盖
- 是否运行 nvidia-xconfig utillty，选择 是
- 完成

• 验证

nvidia-smi

安装 cuda

步骤如下：

• 进入文件夹目录

cd /home/xu/download/tftools

• 给文件加上可执行权限

sudo chmod u+x cuda_10.0.130_410.48_linux.run

• 运行安装文件

sudo ./cuda_10.0.130_410.48_linux.run

注意：
- 首先是一个声明，可以疯狂按着enter键直到过完整个声明；
- do you accept …? accept
- install NVIDIA 什么什么图形驱动…？ n
- install the CUDA 10.0 Toolkit? y
- 后边都是填的 y

• 添加环境变量

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

• 保存并退出

source ~/.bashrc

• 验证
  选择前述安装的一个CUDA例子，运行以下命令来测试是否安装成功：

cd /usr/local/cuda/samples/1_Utilities/deviceQuery 
sudo make
./deviceQuery

有一片看似正常的内容就代表成功安装了。

安装CUDNN

• 解压cudnn安装包
  
• 进入解压后的文件夹

cd cudnn-10.0-linux-x64-v7.4.2.24

• 拷贝文件

sudo cp cuda/include/cudnn.h /usr/local/cuda-10.0/include/ 
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-10.0/lib64/ 

• 更改文件执行权限（保证文件可读就行）

sudo chmod a+r /usr/local/cuda-10.0/include/cudnn.h 
sudo chmod a+r /usr/local/cuda-10.0/lib64/libcudnn*

• 查看CUDNN版本

cat /usr/local/cuda-10.0/include/cudnn.h

文件前几行有用代码中可以看到CUDNN的版本号

参考资料

参考链接地址： jeanhwea tensorflow安装笔记