安裝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即可。

圖上，表示安裝成功啦！！！