1 .查看显卡的驱动版本

帮助-> 系统信息

可知我的是cuda11的驱动
然后就去官网找对应的安装包下载
我选择了本地安装

精简安装
安装完成后测试看看是否安装成功
控制台输入
下载cuDNN
https://developer.nvidia.com/rdp/cudnn-archive
到这网站去寻找对应的版本
安装cuDNN
解压后得到三个文件夹,把这三个文件夹中的内容分别放到之前cuda对应的文件夹中。
2 安装pytorch
pip install torch===1
CUDA驱动与PyTorch安装指南
本文介绍如何检查和安装CUDA驱动,下载并安装cuDNN,以及通过pip安装特定版本的PyTorch、torchvision和torchaudio。文中还提供了安装过程中的详细步骤和命令示例。

帮助-> 系统信息

可知我的是cuda11的驱动
然后就去官网找对应的安装包下载
我选择了本地安装

精简安装
安装完成后测试看看是否安装成功
控制台输入
https://developer.nvidia.com/rdp/cudnn-archive
到这网站去寻找对应的版本
解压后得到三个文件夹,把这三个文件夹中的内容分别放到之前cuda对应的文件夹中。
pip install torch===1
3127
6465
3322