四步解决pytorch-gpu安装 【亲测可用】

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则安装成功。