1.查看软件版本间对应关系
cuda 和显卡驱动关系
torch- torchvision- python版本对应关系
CUDA Toolkit 和Pytorch的对应关系
2.进入该网址找到合适的pytorch和torchvision版本下载https://download.pytorch.org/whl/torch_stable.html
我使用的版本是cuda11.1 python3.7 Windows10(64位)操作系统,那么我应该选择的安装包如下图:
3.下载完两个文件后进入环境进行pip安装。
进入需要安装pytorch—gpu的环境(若没有则创建一个,创建教程看我上一篇文章)
分别输入命令pip install + (下载whl文件的路径)进行安装。
例如我下载的路径是D:\Downloads
则我输入的命令为:
pip install D:\Downloads\torch-1.8.1+cu111-cp37-cp37m-win_amd64.whl、
pip install D:\Downloads\torchvision-0.9.1+cu111-cp37-cp37m-win_amd64.whl
4.安装完成后测试
在该环境下进入python环境输入
import torch
print(torch.cuda.is_available())
若返回Ture则安装成功。