Ubuntu16.04 配置显卡驱动+CUDA10.1+cudnn

基本说明显卡驱动安装CUDA 10.1安装cudnn下载

基本说明

本篇主要基于ubuntu16.04进行gpu环境配置，gpu型号 1050Ti，请各位根据自己的显卡型号各取所需。

显卡驱动安装

首先确认机器是否带有显卡 lspci | grep -i vga

带有N卡可进行下一步的nvidia 驱动安装 这里显示的1e82 设备对应Geforce1050Ti

从官网下载.run进行安装 对于未利用系统扩展方式安装驱动的情况，可以先到官方网站查看最新支持到的驱动版本 nvidia中文官网驱动下载 选择计算机对应的显卡

进行下载，假设保存于Downloads内

禁用nouveau sudo vim /etc/modprobe.d/blacklist.conf

在文件末尾添加

blacklist nouveau 进入命令行界面，准备安装 进行这个操作前，推荐各位记录好之前所做操作，以及准备好一个设备查看后续部分 先ctrl+alt+F1进入非图形化界面，关闭图像化界面 sudo service lightdm stop

赋予可执行权限

cd Downloads sudo chmod a+x NVIDIA-Linux-x86_64-430.50 进行安装 为避免出现桌面循环登录，-no-x-check -no-nouveau-check -no-opengl-files 一定要有禁用 opengl部分 sudo ./NVIDIA-Linux-x86_64-396.18.run -no-x-check -no-nouveau-check -no-opengl-files

进安装状态，有几个选项 [1] Would you like to register the kernel module souces with DKMS? This will allow DKMS to automatically build a new module, if you install a different kernel later? 选择 No [2] Nvidia’s 32-bit compatibility libraries? 选择 No [3] Would you like to run the nvidia-xconfigutility to automatically update your x configuration so that the NVIDIA x driver will be used when you restart x? Any pre-existing x confile will be backed up. 选择 No 安装完毕后，可开启并进入图形界面

sudo service lightdm start

终端内输入

nvidia-smi

可查看到刚刚安装的驱动 至此驱动安装完毕

另一种安装方法（不推荐） 依赖ubuntu系统自身扩展软件安装 关于这一类安装方法的驱动查看

sudo dpkg --list | grep nvidia-*

实际上是走的apt-get install 安装的，依赖于linux的源 当你想要卸载时

sudo apt-get remove --purge nvidia*

即可卸载由此方法安装的nvidia驱动

CUDA 10.1安装

首先先确认一件事情 在官网说明文当中 官网cuda toolkit说明文档 不同驱动版最高可按装的cuda版本不同，会导致最终环境有差别 下面开始cuda驱动安装 cuda官网下载驱动链接 选择好自己机器对应的型号 最后install type选择这里可以根据个人网速状态选择前两个，本文安装时选择了runfile

wget http://developer.download.nvidia.com/compute/cuda/10.1/Prod/local_installers/cuda_10.1.243_418.87.00_linux.run

下载好后，依照官网的安装说明进行安装

sudo sh cuda_10.1.243_418.87.00_linux.run

安装时，先出现是否同意协议的确认，输入accept 由于前面单独安装了最新的显卡驱动，这里 这样子选择即可，X代表要安装的部分 安装完成后，配置~/.bashrc

nano ~/.bashrc

在末尾添加

export PATH=/usr/local/cuda-10.1/bin${PATH:+:${PATH}} export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}} source ~/.bashrc

即可查看刚刚完成安装的cuda版本

nvcc -V

cudnn下载

这里基本上只有下载和拷贝的操作，所以与其说是安装，不如说时环境配置的库添加 首先，在nvidia官网找到cudnn的对应版本，下载linux library *.tgz包 nvidia官网提供的 cudnn 地址 下载好了后，解压

tar -zxvf cudnn-9.0-linux-x64-v7.6.1.34.tgz

复制到cuda目录下

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 /usr/local/cuda/lib64/libcudnn* #加权限

检查

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

这里 cudnn7.6.4的就按装完毕了

参考链接