cuda 与驱动版本对应关系：
方式一：官方安装教程：
下载地址：https://developer.nvidia.com/cuda-downloads?target_os=Linux&target_arch=x86_64&Distribution=Ubuntu&target_version=22.04&target_type=deb_local
Download Installer for Linux Ubuntu 22.04 x86_64
The base installer is available for download below.
Base Installer
Installation Instructions:
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-ubuntu2204.pin
sudo mv cuda-ubuntu2204.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/12.5.0/local_installers/cuda-repo-ubuntu2204-12-5-local_12.5.0-555.42.02-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2204-12-5-local_12.5.0-555.42.02-1_amd64.deb
sudo cp /var/cuda-repo-ubuntu2204-12-5-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get -y install cuda-toolkit-12-5
Additional installation options are detailed here.
Driver Installer
NVIDIA Driver Instructions (choose one option)
To install the legacy kernel module flavor:
sudo apt-get install -y cuda-drivers
To install the open kernel module flavor:
sudo apt-get install -y nvidia-driver-555-open
sudo apt-get install -y cuda-drivers-555
安装完成后配置系统环境变量
sudo vi /etc/profile
在末尾加入以下代码：
export PATH=$PATH:/usr/local/cuda-12.5/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-12.5/lib64
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-12.5
保存退出
系统环境生效：
source /etc/profile
验证版本
nvcc –version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2024 NVIDIA Corporation
Built on Wed_Apr_17_19:19:55_PDT_2024
Cuda compilation tools, release 12.5, V12.5.40
Build cuda_12.5.r12.5/compiler.34177558_0
方式二：终端输入以下命令
sudo apt install nvidia-cuda-toolkit
查看版本：
nvcc –version
查询显卡信息：
nvidia-smi
——————————————————–分割线—————————————————————-
重要！！如果你的模型要用到特定版本的pytorch，记得先看看跟这个版本的pytorch相匹配的cudnn：Previous PyTorch Versions
ubuntu安装cuDNN教程：
cuDNN 存档
NVIDIA cuDNN 是一个用于深度神经网络的 GPU 加速基元库
进入官网https://developer.nvidia.com/rdp/cudnn-archive#a-collapse897-120
选择版本下载
下载时会要求登录，注册个账户登录一下就可以下载了
deb安装包下载链接：
https://developer.download.nvidia.cn/compute/cudnn/secure/8.9.7/local_installers/12.x/cudnn-local-repo-ubuntu2204-8.9.7.29_1.0-1_amd64.deb?vjP8PookKFF3NeKbXQ3mYiJJ5D2qRT6OhbBdBDqTvyqHWllEWcwTp0DyJb5d8XRx-aFpJTnHWiC6Df1nmQDOpUJYbqDW6HBNiTtCJOG6KAMT5iSYaISeZtJAkm8J9tQjb3Te2AfOUVorKHk3jWaBqXSnNtF3p6SjCKVvvyqZV_khCVSIp318pidrL4ncC0Xi2LKz60EovH4DehypmkVtudGPHYU=&t=eyJscyI6IndlYnNpdGUiLCJsc2QiOiJkZXZlbG9wZXIubnZpZGlhLmNvbS9sb2dpbiJ9
然后执行
sudo dpkg -i cudnn-local-repo-ubuntu2204-8.9.7.29_1.0-1_amd64.deb
复制文件到CUDA安装目录
sudo cp /var/cudnn-local-repo-ubuntu2204-8.9.7.29/cudnn-local-08A7D361-keyring.gpg /usr/share/keyrings/
压缩包安装：
https://developer.download.nvidia.cn/compute/cudnn/secure/8.9.7/local_installers/12.x/cudnn-linux-x86_64-8.9.7.29_cuda12-archive.tar.xz?wRWKKYgWB_YhC3ji5Yt6pmlAgbvGZCS1LKcJ9IKrefjlf4dpxZZJ5ljB3Kk9Y5xXnCH4pCiUmR60WLbaT3b2mC8xFURjZsyCYt3njFWaf9fSQsoxaPoWFXMlgjK5o35QTRm4Anz5WmrtueENGlKxkRL7gFWPjF0TEVSDxodeC–bK2bDavU7iZwOcaH3vpoP9WC61ChOiOpHdfNNMZ-AYWs=&t=eyJscyI6IndlYnNpdGUiLCJsc2QiOiJkZXZlbG9wZXIubnZpZGlhLmNvbS9sb2dpbiJ9
将压缩包解压
tar -xvf cudnn-linux-x86_64-8.9.7.29_cuda12-archive.tar.xz
复制文件到CUDA安装目录
sudo cp cudnn-linux-x86_64-8.9.7.29_cuda12-archive/include/* /usr/local/cuda-12.5/include
sudo cp cudnn-linux-x86_64-8.9.7.29_cuda12-archive/lib/libcudnn* /usr/local/cuda-12.5/lib64
添加读取权限
sudo chmod a+r /usr/local/cuda-12.5/include/cudnn.h
sudo chmod a+r /usr/local/cuda-12.5/lib64/libcudnn*
测试
cat /usr/local/cuda-12.5/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
#define CUDNN_MAJOR 8
#define CUDNN_MINOR 9
#define CUDNN_PATCHLEVEL 7
—
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)
/* cannot use constexpr here since this is a C-only file */
复制文件到CUDA安装目录
sudo cp cudnn-linux-x86_64-8.9.7.29_cuda12-archive/include/cudnn.h /usr/local/cuda/include
sudo cp cudnn-linux-x86_64-8.9.7.29_cuda12-archive/lib/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 2
若上述命令没有结果，则运行以下命令
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
若出现以下信息，恭喜你，cudnn安装成功！
#define CUDNN_MAJOR 8
#define CUDNN_MINOR 9
#define CUDNN_PATCHLEVEL 7
—
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)
/* cannot use constexpr here since this is a C-only file */
原创不易请大大们多多支持，扫描支付宝二维码领红包。
构建高质量的技术交流社群，欢迎从事编程开发、技术招聘HR进群，也欢迎大家分享自己公司的内推信息，相互帮助，一起进步！
加我微信拉你进群，备注ittce，不然不加。
微信赞赏
支付宝扫码领红包
