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:
# 安装cuda11对应的cudnn
pip install nvidia-cudnn-cu11
你可以通过以下命令检索可供选择的cudnn版本:
pip index versions nvidia-cudnn-cu11
cuda其他版本同理。