ubuntu安装显卡驱动、cuda、cudnn

一、安装nvidia显卡驱动
查看推荐安装的驱动

ubuntu-drivers devices

找到推荐的驱动版本

Selection_002.png

安装驱动

sudo apt install nvidia-driver-460

二、安装cuda
下载对应cuda toolkit
https://developer.nvidia.com/cuda-downloads
选择安装版本

Selection_004.png

wget https://developer.download.nvidia.com/compute/cuda/11.4.0/local_installers/cuda_11.4.0_470.42.01_linux.run
sudo sh cuda_11.4.0_470.42.01_linux.run

安装过程中跳过驱动安装
三、cudnn
下载https://developer.nvidia.com/rdp/cudnn-download
下载完成后执行

tar -xzvf 刚才下载cudnn压缩包名称

sudo cp cuda/include/cudnn*.h /usr/local/cuda/include 
sudo cp -P 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_version.h | grep CUDNN_MAJOR -A 2