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

相关推荐
剑穗挂着新流苏3121 小时前
115_PyTorch 实战:从零搭建 CIFAR-10 完整训练与测试流水线
人工智能·pytorch·深度学习·神经网络
Fleshy数模9 小时前
基于PyTorch的食品图像分类:数据增强与调优实战
人工智能·pytorch·分类
chushiyunen15 小时前
大模型.safetensors文件
人工智能·pytorch·深度学习
剑穗挂着新流苏31216 小时前
116_深度学习提速:PyTorch 利用 GPU 训练的全攻略
人工智能·pytorch·深度学习·神经网络
Zzzz_my1 天前
正则表达式(RE)
pytorch·python·正则表达式
alex18011 天前
pytorch LSTM类解析
pytorch·机器学习·lstm
剑穗挂着新流苏3121 天前
114_PyTorch 进阶:模型保存与读取的两大方式及“陷阱”避坑指南
人工智能·pytorch·深度学习
GoCoding1 天前
Triton + RISC-V
pytorch·openai·编译器
GoCodingInMyWay1 天前
Triton + RISC-V
pytorch·riscv·triton
rebekk1 天前
pytorch custom op的简单介绍
人工智能·pytorch·python