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

相关推荐
人有一心7 小时前
深度学习里的树模型TabNet
人工智能·深度学习
强盛小灵通专卖员7 小时前
边缘计算设备NPU的加速原理
人工智能·深度学习·边缘计算·sci·中文核心·小论文
moz与京7 小时前
【面试向】边缘计算基础介绍
人工智能·边缘计算
ShiMetaPi7 小时前
【ShiMetaPi】基于BM1684X的智能工业视觉边缘计算盒子解决方案
人工智能·边缘计算·bm1684x
强盛小灵通专卖员7 小时前
RK3576边缘计算设备部署YOLOv11
人工智能·深度学习·yolo·边缘计算·sci·rk3576·小论文
Ms_Big7 小时前
ppliteseg改rknn,部署在嵌入式板,加速模型
人工智能·python·深度学习
说私域7 小时前
基于开源AI智能名片链动2+1模式S2B2C商城小程序的移动互联网人气氛围营造机制研究
人工智能·小程序·开源
TMT星球7 小时前
从IFA再出发:中国制造与海信三筒洗衣机的“答案”
人工智能·制造
edisao7 小时前
[特殊字符] 从助手到引擎:基于 GPT 的战略协作系统演示
大数据·人工智能·gpt
三之又三8 小时前
卷积神经网络CNN-part5-NiN
人工智能·神经网络·cnn