正文
Ubuntu18.04上安装N卡驱动、CUDA、CUDNN三连
提示：扫一扫查出行【扫一扫了解最新限行尾号】
复制提示
环境：Ubuntu18.04
显卡驱动真的挺方便的，CUDA和CUDNN还是踩了一些坑2333
1、安装显卡驱动
安装ubuntu更新或sudo apt-get update & sudo apt-get upgrade
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
ubuntu-drivers devices
sudo ubuntu-drivers autoinstall
安装显卡驱动
sudo reboot
nvidia-smi
若输出显卡信息则安装成功
非常方便了
2、安装CUDA
在官网https://developer.nvidia.com/cuda-downloads?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1804&target_type=runfilelocal下载安装包，默认下载到Downloads目录下
cd ~/Downloads/
运行下载的.run文件，如：cuda_10.1.105_418.39_linux.run
sudo sh cuda_*_linux.run
在过程中，选择需要安装的内容时，请将驱动driver去掉，不要安装显卡驱动！
配置环境变量（该步骤似乎并不需要，没有进行该操作也成功运行了后续测试，有待考证）
export CUDA_HOME=/usr/local/cuda
export PATH=$PATH:$CUDA_HOME/bin
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
source ~/.bashrc
sudo reboot
测试是否安装成功，这里如果没有重启过，貌似运行的时候会有报错
cd /usr/local/cuda/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery
查看CUDA版本
cat /usr/local/cuda/version.txt
3、安装CUDNN
在官网https://developer.nvidia.com/cudnn注册账号并下载，选择下载cuDNN Library for Linux，不过说实话现在官网给的版本好多好多看花了眼23333，默认下载到Downloads目录下，解压并复制
cd ~/Downloads/
tar -xzvf cudnn-*
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*
查看CUDNN版本
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A
