1. win+r cmd 进入命令窗口行 输入 nvcc --version 找cuda版本 如果没有安装cuda自行搜索安装
2.https://download.pytorch.org/whl/torch_stable.htm
在以上网址中寻找cuda版本对应的torch版本
以cuda==11.5 python==3.88举例 对应文件名:cu115 cp38 看清win 还是 linux操作系统
下载三个文件:1)torch 2) torchaudio 3) torchvision
3.将以上三个文件保存到项目下面,(注意不是site-packages下面,我犯过错)
例如:e:\pythonproject\dann_py3-master
4.在终端pip install torch-1.11.0+cu115-cp38-cp38-win_amd64.whl(根据自己下载的版本更改)
5.其他补充:
将之前下载过的 torch torchaudio torchvision都 uninstall