ubuntu22.04 安装cuda cudnn

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 cudnnhttps://blog.csdn.net/m0_53931365/article/details/146322464?spm=1001.2014.3001.5502

bash 复制代码
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

bash 复制代码
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

相关推荐
aigcapi1 小时前
RAG 系统的黑盒测试:从算法对齐视角解析 GEO 优化的技术指标体系
大数据·人工智能·算法
上进小菜猪1 小时前
基于深度学习的河道垃圾检测系统设计(YOLOv8)
人工智能
上天夭2 小时前
模型训练篇
人工智能·深度学习·机器学习
小徐Chao努力2 小时前
【Langchain4j-Java AI开发】09-Agent智能体工作流
java·开发语言·人工智能
Blossom.1182 小时前
AI编译器实战:从零手写算子融合与自动调度系统
人工智能·python·深度学习·机器学习·flask·transformer·tornado
Coder_Boy_2 小时前
SpringAI与LangChain4j的智能应用-(理论篇2)
人工智能·spring boot·langchain·springai
却道天凉_好个秋2 小时前
OpenCV(四十八):图像查找
人工智能·opencv·计算机视觉
Coder_Boy_2 小时前
SpringAI与LangChain4j的智能应用-(理论篇3)
java·人工智能·spring boot·langchain
GetcharZp2 小时前
工地“火眼金睛”!手把手带你用 YOLO11 实现安全帽佩戴检测
人工智能·计算机视觉
Codebee2 小时前
Ooder A2UI架构白皮书
人工智能·响应式编程