Windows上安裝:html
最新0.4.0版本:python
在Pytorch官網 https://pytorch.org/ 上選擇對應版本安裝便可,conda安裝的比較慢,建議選擇pip安裝的(雖然仍是很慢),固然能找到不錯的鏡像也是極好的。安裝CPU版的在CUDA處選None。linux
0.3.0等之前的老版本:post
建議參考知乎https://www.zhihu.com/question/67209417,用裏面的鏡像會快一些。測試
Linux上安裝:url
先裝好CUDA和Cudnn,參考 https://www.cnblogs.com/sbj123456789/p/9219631.htmlspa
建立環境:.net
conda create -n pytorch python=3.6htm
激活環境:blog
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上會不支持。