Linux/Windows GPU/CPU版 pytorch安裝

Windows上安裝：

最新0.4.0版本：

在Pytorch官網 https://pytorch.org/ 上選擇對應版本安裝即可，conda安裝的比較慢，建議選擇pip安裝的（雖然還是很慢），當然能找到不錯的鏡像也是極好的。安裝CPU版的在CUDA處選None。

0.3.0等以前的老版本：

建議參考知乎https://www.zhihu.com/question/67209417，用里面的鏡像會快一些。

Linux上安裝：

先裝好CUDA和Cudnn，參考 https://www.cnblogs.com/sbj123456789/p/9219631.html

創建環境：

　　conda create -n pytorch python=3.6

激活環境：

　　conda activate pytorch 或 source activate pytorch

退出環境：

　　conda deactivate

查看所有環境：

　　conda env list

直接pip install http://download.pytorch.org/whl/cu80/torch-0.2.0.post3-cp36-cp36m-manylinux1_x86_64.whl

（直接pip install torch torchvision慢出天際）

把后面直接換成想要的版本即可，具體可參考官網。

GPU安裝后的測試：

換成清華源：

直接從國外的源下載速度還是不夠快，可以換成清華源。

參考：https://blog.csdn.net/zzq060143/article/details/88042075

pytorch和CUDA版本號改一下就ok

一些坑：

GPU 1080 以下的版本盡量安裝0.3.0的，安裝0.4.0的在Windows上會不支持。