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其他版本同理。

相关推荐
fantasy_arch10 小时前
BasicVSR-lite图像画质增强
开发语言·pytorch
Will_Ye12 小时前
Ubuntu中pip 缓存和conda的缓存路径修改
缓存·conda·pip
小a杰.19 小时前
Torchtitan NPU 框架 - PyTorch与NPU集成指南
人工智能·pytorch·python
盼小辉丶20 小时前
PyTorch强化学习实战——使用高级组件复现DQN
pytorch·深度学习·强化学习
徐安安ye1 天前
FlashAttention流式输出:Streaming Chunked Attention与增量解码
人工智能·pytorch·python
SimpleLearingAI2 天前
PyTorch乘法全解析
人工智能·pytorch·python
MediaTea3 天前
PyTorch:主要模块简介
人工智能·pytorch·python·深度学习·机器学习
rayyy93 天前
神经网络模型的外推性验证
pytorch·python·深度学习
MediaTea3 天前
DL:扩散模型的基本原理与 PyTorch 实现
人工智能·pytorch·python·深度学习·机器学习
EnCi Zheng3 天前
09aa-偏置是什么?
人工智能·pytorch·python·深度学习·神经网络