Pytorch2.4.0自动安装cudnn9.1??? pip安装cudnn方法

Pytorch2.4.0使用官方安装方式安装的时候,会自动给我安装cudnn9.1,不论是conda安装还是pip安装,害得我出了一大堆问题,气死我了

我倒退到安装pytorch2.3.0才没出现自动安装cudnn的问题

以及,记录一个pip安装cudnn的方法,这也是官方推荐的https://docs.nvidia.com/deeplearning/cudnn/latest/installation/linux.html:

shell 复制代码
# 安装cuda11对应的cudnn
pip install nvidia-cudnn-cu11

你可以通过以下命令检索可供选择的cudnn版本:

shell 复制代码
pip index versions nvidia-cudnn-cu11

cuda其他版本同理。

相关推荐
Yolanda_20222 小时前
PyTorch 线性层nn.Linear底层数学原理
人工智能·pytorch·python
专注仿真2 小时前
多智能体强化学习实战:从 Dec-POMDP 到 QMIX/MADDPG/MAPPO 的可运行 PyTorch 实现
人工智能·pytorch·python
Web3&Basketball15 小时前
TorchTPU 迁移实战:3 个 CUDA 卡点
pytorch·性能调优·cuda·tpu·模型迁移·torchtpu
田里的水稻1 天前
FA_CNN卷积神经网络十二_CIFAR10全流程PyTorch
人工智能·pytorch·深度学习·神经网络·cnn
论文复现现场1 天前
RTX 3090 云GPU租赁价格对比:8家平台配置、计费方式与环境测试
linux·运维·人工智能·pytorch·深度学习·云计算·3090
liwulin05061 天前
【PYTORCH】迷你DECODER-ONLY模型
pytorch·python·深度学习
田里的水稻2 天前
FA_CNN卷积神经网络四---线性层及其拉平PyTorch
人工智能·pytorch·python·深度学习·神经网络·cnn
GPU实战笔记2 天前
NVIDIA L40S 云端任务验证:训练、推理与生成工作流怎样先试再租
pytorch·ai生成·gpu租用·nvidia l40s
QN1幻化引擎2 天前
信号场注意力(SFA):把 KV Cache 变成 O(1) 的双通道注意力——0.1B/0.3B 受控 A/B 精度反升 15.5% / 24.3%
pytorch·深度学习·大模型·transformer·注意力机制·推理优化
田里的水稻2 天前
FA_CNN卷积神经网络二---PyTorch
人工智能·pytorch·python·深度学习·cnn