使用pip+代理下载PyTorch GPU版本（win10系统）

• 前言
• 查看PyTorch官网
• 查看代理服务器地址、端口
• 使用pip下载
• 可能出现的错误
• • 1. `ReadTimeoutError`
  • 2. `zipfile.BadZipFile: File is not a zip file`
• 成功图片和测试代码

本人是新手，如有错误还请指正。
本文章建立在已知需要下载的PyTorch版本、如何使用代理的基础上。
电脑是Win10系统，conda环境也可以使用pip下载。因为直接下载速度非常缓慢，考虑使用代理来提高下载速度。

首先，需要查看官网确认自己需要下载的PyTorch版本的命令：

pip install torch==1.8.2+cu111 torchvision==0.9.2+cu111 torchaudio===0.8.2 -f https://download.pytorch.org/whl/lts/1.8/torch_lts.html

可以在代理软件的设置中找到，之后的代码请根据你自己的地址和端口进行修改！

需要使用pip/pip3的--proxy选项：
127.0.0.1:41091是我的代理服务器地址和端口，请根据你自己的地址和端口进行修改！

pip3 --proxy 127.0.0.1:41091 install torch==1.8.2+cu111 torchvision==0.9.2+cu111 torchaudio===0.8.2 -f https://download.pytorch.org/whl/lts/1.8/torch_lts.html

超时出错，考虑加大超时时间（在以上代码中加入--default-timeout=1000）：

pip3 --default-timeout=1000 --proxy 127.0.0.1:41091 install torch==1.8.2+cu111 torchvision==0.9.2+cu111 torchaudio===0.8.2 -f https://download.pytorch.org/whl/lts/1.8/torch_lts.html

这个问题比较奇怪，目前我没有发现较好的解决方法。可能是下载文件出错导致的，和网络稳定性有关，只能重试一次。

图片如下：

测试代码：
转自：https://blog.csdn.net/luoshiyong123/article/details/107221694/

import torch
# 以下代码只有在PyTorch GPU版本上才会执行
import time

print(torch.__version__)
print(torch.cuda.is_available())
a = torch.randn(10000, 1000)
b = torch.randn(1000, 2000)
t0 = time.time()
c = torch.matmul(a, b)
t1 = time.time()
print(a.device, t1 - t0, c.norm(2))

device = torch.device('cuda')
a = a.to(device)
b = b.to(device)
t0 = time.time()
c = torch.matmul(a, b)
t1 = time.time()
print(a.device, t1 - t0, c.norm(2))

t0 = time.time()
c = torch.matmul(a, b)
t1 = time.time()
print(a.device, t1 - t0, c.norm(2))

结果为：