ubuntu20.0 新系统 安装N卡驱动、CUDA和CUDNN

前期工作

sudo apt update
sudo apt install openssh-server
sudo apt install vim
sudo apt install net-tools 

1、

查看显卡型号

lspci | grep -i nvidia

安装 gcc、g++、make、核

sudo apt-get install build-essential
sudo apt-get install linux-headers-$(uname -r)
sudo apt-get install freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev

下载驱动、CUDA、Cudnn

2、禁用NOUVEAU

具体操作可参考此博客

lsmod | grep nouveau

若没有输出，则为禁用完成。禁用完后重启。
注意，如果在sudo service lightdm stop 时报unit lightdm.service not loaded 则先 执行sudo apt install lightdm，然后再sudo service lightdm stop

3、安装N卡驱动

sudo chmod a+x NVIDIA-Linux-x86_64-455.45.01.run
sudo ./NVIDIA-Linux-x86_64-455.45.01.run -no-x-check -no-nouveau-check -no-opengl-files

安装完成后，输入nvidia-smi测试，然后重启。
重启后，开机不显示界面而是在最顶端显示 /dev/sda2: clean, xxxx/xxxxxx files , xxxx/xxxxxx blocks
先进入命令行模式 Ctrl+Alt+F1 （F1~F6都可）
然后输入：

rm -rf /etc/X11/xorg.com
cp /etc/X11/xorg.conf.failsafe /etc/X11/xorg.conf

重启。ps：X11的X大写，其他小写。
卸载驱动：

sudo /usr/bin/nvidia-uninstall
sudo apt-get --purge remove nvidia-*
sudo apt-get --purge remove xserver-xorg-video-nouveau

4、安装cuda

sudo sh cuda_11.0.3_450.51.06_linux.run

安装完后

sudo vim ~/.bashrc
export PATH=/usr/local/cuda-11.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

刷新source ~/.bashrc
终端中输入 nvcc -V，会显示cuda信息

5、安装cudnn

解压cudnn压缩包后，

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

或者

sudo dpkg -i libcudnn8_8.0.5.39-1+cuda11.0_amd64.deb
sudo dpkg -i libcudnn8-dev_8.0.5.39-1+cuda11.0_amd64.deb 
sudo dpkg -i libcudnn8-samples_8.0.5.39-1+cuda11.0_amd64.deb 

验证cudnn

sudo cp -r /usr/src/cudnn_samples_v8/ /home/cudnntest
cd /home/cudnntest/mnistCUDNN
sudo make

 sudo ./mnistCUDNN

最后出现 Test passed! 即可