安装 显卡驱动 + cuda + cudnn (笔记)

参考官网 https://docs.nvidia.com/deeplearning/sdk/cudnn-install/index.html

① 安装显卡驱动 Drivers

第一步

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

第二步 查看支持安装的显卡驱动版本号

ubuntu-drivers devices

driver : nvidia-driver-410 - third-party free
driver : nvidia-driver-418 - third-party free
driver : nvidia-driver-415 - third-party free
driver : nvidia-driver-430 - third-party free recommended
driver : xserver-xorg-video-nouveau - distro free builtin

第三步： 选择自动安装

sudo ubuntu-drivers autoinstall

下载完后需要设置重启之后的验证密码，设置完后重启
界面跳出
reboot
enroll from mok
enroll key from disk
enroll hash from disk
选择 enroll from mok
选择 yes – reboot
第二种: Software&Updates
在 Additional Drivers 里面找到 (recommend)的驱动 , apply
完成后也是需要设置密码重启,
同上操作
nvidia-smi 查看结果

②安装 cuda toolkit

1. 官网下载完，按照下载页面下的操作代码安装即可

2. 给cuda添加环境变量
   sudo nano ~/.bashrc
   最后行添加
   export PATH=/usr/local/cuda/binKaTeX parse error: Expected '}', got 'EOF' at end of input: {PATH:+:{PATH}}
   export LD_LIBRARY_PATH=/usr/local/cuda/lib64KaTeX parse error: Expected '}', got 'EOF' at end of input: …LIBRARY_PATH:+:{LD_LIBRARY_PATH}}
   export CUDA_HOME=/usr/local/cuda

source ~/.bashrc 使环境变量生效

③ 安装cudnn
首先找到对应之前装的cuda的cudnn版本
因为我是在Ubuntu18.04安装,所以分别下载 for ubuntu 18.04这三个文件
然后开始安装三个文件 sudo dpkg -i <分别是刚刚下的三个文件,少一个都不行>
测试结果
#Copy the cuDNN sample to a writable path.
cp -r /usr/src/cudnn_samples_v7/ ./

#Go to the writable path.
cd ./cudnn_samples_v7/mnistCUDNN

这个有个bug坑,如果直接下一步make会编译失败,cd 到 /usr/include/ 下面把 cudnn.h 文件里面的
#include “driver_types.h”
改成:
#include
这样才能下一步编译成功

#Compile the mnistCUDNN sample.
make clean && make

#Run the mnistCUDNN sample.
./mnistCUDNN

If cuDNN is properly installed and running on your Linux system, you will see a message similar to the following:
Test passed!