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

相关推荐
龙文浩_3 小时前
AI深度学习中的PyTorch与张量案例
人工智能·pytorch·深度学习
Storynone13 小时前
【踩坑笔记】Geforce RTX5060 显卡对应的 Pytorch 安装
人工智能·pytorch·笔记
我材不敲代码13 小时前
NLP 入门——PyTorch 从零实现 Word2Vec 之 CBOW 模型
pytorch·自然语言处理·word2vec
致宏Rex15 小时前
uv 教程:安装、常用命令、项目结构与关键文件
python·pip·uv
郝学胜-神的一滴15 小时前
PyTorch张量维度操控:transpose与permute深度拆解与实战指南
人工智能·pytorch·python·深度学习·算法·机器学习
星川皆无恙15 小时前
Transformer学习笔记:从 Attention 核心理论到机器翻译代码项目学习实战
大数据·pytorch·笔记·深度学习·神经网络·自然语言处理·transformer
dazzle18 小时前
机器学习算法原理与实践-入门(十一):基于PyTorch的房价预测实战
pytorch·算法·机器学习
白小筠1 天前
Pytorch之张量的基本操作
人工智能·pytorch·python
热爱生活的猴子2 天前
PyTorch导出ONNX报错(ShapeInferenceError)问题笔记(含dynamo=False作用解析)
人工智能·pytorch·笔记
芜湖xin2 天前
【解决Error】pip安装Flask失败
python·flask·pip