ubuntu22.04 安装cuda cudnn

发布于:2025-07-07 ⋅ 阅读:(16) ⋅ 点赞:(0)

1.输入nvidia-smi查看可以支持安装的cuda最大版本

2.cuda与cudnn版本的选择

核心原则

  1. 向下兼容性:较新的 cuDNN 通常兼容旧版 CUDA,但反之不成立

  2. 框架依赖:优先考虑深度学习框架(TensorFlow/PyTorch)的版本要求

  3. 硬件限制:GPU 架构决定支持的最高 CUDA 版本

详细选择策略

2.1根据深度学习框架选择
框架 推荐配置 验证命令
PyTorch CUDA 12.x + cuDNN ≥8.9 python -c "import torch; print(torch.version.cuda)"
TensorFlow CUDA 12.x + cuDNN ≥8.9 python -c "import tensorflow as tf; print(tf.sysconfig.get_build_info())"
MXNet CUDA 11.8 + cuDNN 8.6 nvidia-smi 查看驱动版本
2.2 根据 GPU 架构选择
GPU 架构 最大支持 CUDA 推荐组合
Ada (RTX 40系) 12.4+ CUDA 12.4 + cuDNN 9.1
Ampere (A100/A40) 12.x CUDA 12.3 + cuDNN 8.9
Turing (RTX 20系) 11.8 CUDA 11.8 + cuDNN 8.7
Pascal (GTX 10系) 11.0 CUDA 11.0 + cuDNN 8.0

⚠️ 注意:RTX 30/40系显卡必须使用 CUDA 11.1+,旧版本无法识别

3.安装方式参考之前发的windows安装cuda cudnn

我选择12.4cuda 和9.1cudnn(提前安装好显卡驱动)

windows 安装anaconda、cuda、cudnn_anaconda cuda cudnn-CSDN博客文章浏览阅读2.3k次,点赞71次,收藏45次。给自己游戏本配置环境时,发现之前的文章收费了QAQ,打算自己写一篇,方便参考_anaconda cuda cudnn https://blog.csdn.net/m0_53931365/article/details/146322464?spm=1001.2014.3001.5502

 

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.4.0/local_installers/cuda-repo-ubuntu2204-12-4-local_12.4.0-550.54.14-1_amd64.deb

sudo dpkg -i cuda-repo-ubuntu2204-12-4-local_12.4.0-550.54.14-1_amd64.deb

sudo cp /var/cuda-repo-ubuntu2204-12-4-local/cuda-*-keyring.gpg /usr/share/keyrings/

sudo apt-get update

sudo apt-get -y install cuda-toolkit-12-4

CUDA Toolkit 12.4 Downloads | NVIDIA Developerhttps://developer.nvidia.com/cuda-12-4-0-download-archive?target_os=Linux&target_arch=x86_64&Distribution=Ubuntu&target_version=22.04&target_type=runfile_local

wget https://developer.download.nvidia.com/compute/cudnn/9.1.0/local_installers/cudnn-local-repo-ubuntu2204-9.1.0_1.0-1_amd64.deb

sudo dpkg -i cudnn-local-repo-ubuntu2204-9.1.0_1.0-1_amd64.deb

sudo cp /var/cudnn-local-repo-ubuntu2204-9.1.0/cudnn-*-keyring.gpg /usr/share/keyrings/

sudo apt-get updatesudo apt-get -y install cudnn

cuDNN 9.1.0 Downloads | NVIDIA Developerhttps://developer.nvidia.com/cudnn-9-1-0-download-archive?target_os=Linux&target_arch=x86_64&Distribution=Ubuntu&target_version=22.04&target_type=deb_local

 

 


网站公告

今日签到

点亮在社区的每一天
去签到