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

一、安装显卡驱动

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

1. 查看版本对应

在cuda说明文档中查看需要的cuda对应的显卡驱动版本
ubuntu16.04安装显卡驱动+cuda+cudnn_第1张图片
cuda10.1 在 linux 下需要显卡驱动版本 >=418.39

2. 在线安装显卡驱动

  • 添加ppa源
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt update
  • 查看添加的源中支持当前显卡的驱动版本
ubuntu-drivers devices

ubuntu16.04安装显卡驱动+cuda+cudnn_第2张图片

  • 安装需要的显卡驱动,以 nvidia-430 为例,
sudo apt install nvidia-430 

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

  • 查看显卡驱动信息
nvidia-smi

ubuntu16.04安装显卡驱动+cuda+cudnn_第3张图片

二、安装cuda

1.下载

去cudal工具包存档下载需要的 cuda 版本
ubuntu16.04安装显卡驱动+cuda+cudnn_第4张图片

  • 选择第一种 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

ubuntu16.04安装显卡驱动+cuda+cudnn_第5张图片
选择 continue 继续

ubuntu16.04安装显卡驱动+cuda+cudnn_第6张图片
下边输入 accept 继续

ubuntu16.04安装显卡驱动+cuda+cudnn_第7张图片
去掉 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

ubuntu16.04安装显卡驱动+cuda+cudnn_第8张图片

三、安装cudnn

1.通过 cuDNN Library for Linux 安装

  • 在cudnn官网下载所需的 cudnn,需注册登录
    ubuntu16.04安装显卡驱动+cuda+cudnn_第9张图片
    选择 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 版本不同,使用的命令不同,参考链接

ubuntu16.04安装显卡驱动+cuda+cudnn_第10张图片

2.通过 deb 文件进行安装

参考链接

你可能感兴趣的:(cuda,linux,ubuntu,深度学习)