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

相关推荐
烟锁池塘柳0几秒前
【大模型】解码策略:Greedy Search、Beam Search、Top-k/Top-p、Temperature Sampling等
人工智能·深度学习·机器学习
盼小辉丶15 分钟前
PyTorch实战(14)——条件生成对抗网络(conditional GAN,cGAN)
人工智能·pytorch·生成对抗网络
Allen_LVyingbo1 小时前
数智读书笔记系列035《未来医疗:医疗4.0引领第四次医疗产业变革》
人工智能·经验分享·笔记·健康医疗
zzc9211 小时前
时频图数据集更正程序,去除坐标轴白边及调整对应的标签值
人工智能·深度学习·数据集·标签·时频图·更正·白边
isNotNullX1 小时前
什么是数据分析?常见方法全解析
大数据·数据库·数据仓库·人工智能·数据分析
riveting1 小时前
明远智睿H618:开启多场景智慧生活新时代
人工智能·嵌入式硬件·智能硬件·lga封装·3506
夜阑卧听风吹雨,铁马冰河入梦来1 小时前
Spring AI 阿里巴巴学习
人工智能·学习·spring
c7692 小时前
【文献笔记】Automatic Chain of Thought Prompting in Large Language Models
人工智能·笔记·语言模型·论文笔记
Blossom.1182 小时前
机器学习在智能供应链中的应用:需求预测与物流优化
人工智能·深度学习·神经网络·机器学习·计算机视觉·机器人·语音识别
Gyoku Mint2 小时前
深度学习×第4卷:Pytorch实战——她第一次用张量去拟合你的轨迹
人工智能·pytorch·python·深度学习·神经网络·算法·聚类