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

相关推荐
creator_Li4 小时前
深度学习 学习笔记
pytorch·深度学习
派大_星5 小时前
基于 PyTorch 的 CNN 图片分类学习
pytorch·分类·cnn
zhangzeyuaaa6 小时前
深入理解 pip install .:安装当前目录项目的正确姿势
人工智能·python·pip
RobinDevNotes9 小时前
模型训练工程师必须搞懂的DDP和FSDP
人工智能·pytorch
洛阳纸贵9 小时前
AI-PyTorch(一)基础代码实操和自动求导
人工智能·pytorch·python
磁场转动100万匹11 小时前
PyTorch 食物图像分类实战:从数据准备到 CNN 训练全流程解析
pytorch·分类·cnn
ai小陈11 小时前
PyTorch梯度累积与裁剪实战:小显存也能稳定训练大批次
人工智能·pytorch·python·深度学习·ai·gpu算力
C^h20 小时前
pytorch 适合初学者 0基础学习
人工智能·pytorch·python
论文复现现场1 天前
本地 PyTorch 训练 OOM,第一次租 RTX 4090 云 GPU 怎么迁移项目?从环境检查到 100 Step 跑通
人工智能·pytorch·python·深度学习·cuda
2601_962099081 天前
在Python中使用LSTM和PyTorch进行时间序列预测
pytorch·机器学习·lstm·时间序列预测·数据预处理