Linux/Windows GPU/CPU版 pytorch安裝

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上會不支持。

相關文章
相關標籤/搜索