安装GPU版本pytorch教程（避坑）

Pytorch官方网站：

https://pytorch.org/

复制上述语句，执行，但速度超级超级慢。即便有梯子，速度依旧很慢。容易超时失败！

遇到好多好多坑之后，决定采用阿里云的源文件，即在上述代码后附上以下代码：

-i http://mirrors.aliyun.com/pypi/simple/  --trusted-host mirrors.aliyun.com

相当于，用阿里云的pip源并且信任这个源，可以用。

 

具体安装如下：（最好管理员身份打开）

1.更新Pip

python -m pip install --upgrade pip -i http://mirrors.aliyun.com/pypi/simple/  --trusted-host mirrors.aliyun.com

2.安装torch 1.1.0

pip install https://download.pytorch.org/whl/cu90/torch-1.1.0-cp37-cp37m-win_amd64.whl  -i http://mirrors.aliyun.com/pypi/simple/  --trusted-host mirrors.aliyun.com

（注：本案例python版本为3.7，其他版本如python3.6，则修改成cp36-cp36m）

3.安装torchvision 0.3.0

pip3 install https://download.pytorch.org/whl/cu90/torchvision-0.3.0-cp37-cp37m-win_amd64.whl  -i http://mirrors.aliyun.com/pypi/simple/  --trusted-host mirrors.aliyun.com

（注意事项同上）

 

测试torch能否用GPU

>>>import torch

>>>torch.__version__

>>>torch.cuda.is_available()

输出True即可。

图上，表示安装成功啦！！！