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

相关推荐
田里的水稻1 小时前
FA_CNN卷积神经网络十八---感知机到多层感知机PyTorch
人工智能·pytorch·python
听风吹等浪起2 小时前
第27章:EfficientNet全家桶|钢材表面6类缺陷分类完整Pytorch实现
pytorch·分类·数据挖掘
Ivanqhz3 小时前
MoE + LLM 应用
数据库·人工智能·pytorch·python·深度学习·算法·1024程序员节
tellmewhoisi1 天前
深度学习26:PyTorch:参数更新方法(Momentum)
人工智能·pytorch·深度学习
Zack_Liu1 天前
code样例
pytorch·python·深度学习
论文复现现场1 天前
BEVFormer训练需要几张GPU?2张A100 80GB够不够,显存与云GPU成本分析
pytorch·自动驾驶·a100·bevformer
沧海一笑-dj2 天前
【Python】Python学习笔记-模块与包
python·pip·模块·包·python包·python模块·包与模块
Be for thing2 天前
PyTorch基础|Tensor张量详解,张量创建、数据类型与常用操作
pytorch·学习·1024程序员节
地平线开发者2 天前
QAT 实战:从 PyTorch 模型到 J6E 实板精度与延时验证
pytorch·算法·自动驾驶