Pytorch2.4.0自动安装cudnn9.1？？？ pip安装cudnn方法

BigYouYou2024-09-16 18:50

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