ubuntu16.04安装显卡驱动+cuda+cudnn

一、安装显卡驱动

确认需要的 cuda 版本，以 cuda10.1 为例

1. 查看版本对应

在cuda说明文档中查看需要的cuda对应的显卡驱动版本

cuda10.1 在 linux 下需要显卡驱动版本 >=418.39

2. 在线安装显卡驱动

• 添加ppa源

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt update

• 查看添加的源中支持当前显卡的驱动版本

ubuntu-drivers devices

• 安装需要的显卡驱动，以 nvidia-430 为例，

sudo apt install nvidia-430 

注意驱动名字，要与上图中需要的版本一致

• 查看显卡驱动信息

nvidia-smi

二、安装cuda

1.下载

去cudal工具包存档下载需要的 cuda 版本

• 选择第一种 runfile(local) 方式(其他方法也可以)
• 可以通过终端输入 wget 命令下载安装包

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

• 也可以复制上图选中的链接，手动下载安装包

2.安装

• 运行下载打 .run 文件进行安装

sudo sh cuda_10.1.243_418.87.00_linux.run

选择 continue 继续

下边输入 accept 继续

去掉 Driver 前面的 X，不安装 Driver，因为之前以及安装好了，选择 install 开始安装

3.添加环境变量

• 打开 .banshrc 文件

sudo gedit ~/.bashrc

• 在打开的文档最后添加下边三行，最后一行修改为自己的 cuda 版本

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

• 保存文件后退出，终端输入

source ~/.bashrc

4.测试是否安装成功

• 查看cuda版本两种方法，注意空格和大写 V

nvcc -V

cat /usr/local/cuda/version.txt

显示成功安装 cuda10.1.243

• 也可以用下面方法测试

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

三、安装cudnn

1.通过 cuDNN Library for Linux 安装

• 在cudnn官网下载所需的 cudnn，需注册登录
  
  选择 cuDNN Library for Linux (x86)
• 解压下载好的文件，下载好的文件名

tar -xzvf cudnn-10.1-linux-x64-v8.0.5.39.tgz

• 复制文件

sudo cp -r cuda/include/ /usr/local/cuda/include/ 
sudo cp -r cuda/lib64/ /usr/local/cuda/lib64/ 

• 赋予权限

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

• 查看 cudnn 版本

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

或者

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

根据 cudnn 版本不同，使用的命令不同，参考链接

2.通过 deb 文件进行安装

参考链接