win+r输入cmd打开命令行窗口
NVIDIA-smi
查看自己的Driver Version ,CUDA Version
下载whl
查找对应的版本,比如我的是CUDA Version:11.4
那就Ctrl+F搜小于等于114的版本下载(cu113代表CUDA Version:11.3,cp36代表python3.6,amd64代表64位操作系统)
分别下载torch、torchvision、torchaudio
cmd进入whl文件下载页面执行(我的文件在D:\新建文件夹)
请根据自己下载的对应文件进行安装
pip install 文件名
例如
把那三个文件安装了之后输入python
然后输入
import torch
没有报错即完成