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

相关推荐
吴佳浩6 小时前
什么是算力?
人工智能·pytorch·llm
数据智能老司机2 天前
PyTorch 深度学习——使用神经网络来拟合数据
pytorch·深度学习
数据智能老司机2 天前
PyTorch 深度学习——用于图像的扩散模型
pytorch·深度学习
数据智能老司机2 天前
PyTorch 深度学习——Transformer 是如何工作的
pytorch·深度学习
数据智能老司机3 天前
PyTorch 深度学习——使用张量表示真实世界数据
pytorch·深度学习
数据智能老司机3 天前
PyTorch 深度学习——它始于一个张量
pytorch·深度学习
Narrastory5 天前
明日香 - Pytorch 快速入门保姆级教程(三)
pytorch·深度学习
Narrastory8 天前
明日香 - Pytorch 快速入门保姆级教程(一)
人工智能·pytorch·深度学习
Narrastory8 天前
明日香 - Pytorch 快速入门保姆级教程(二)
人工智能·pytorch·深度学习
tryCbest13 天前
Windows环境下配置pip镜像源
windows·pip