1. 程式人生 > >Windows環境下安裝pytorch

Windows環境下安裝pytorch

PyTorch本身主要構件是張量 - 和NumPy看起來差不多。使得PyTorch可以支援大量相同的API,有時候可以把它的用作是NumPy的替代品.PyTorch的開發者們這樣做的原因是希望這種框架可以完全獲得GPU加速帶來的便利,以便你可以快速進行資料預處理,或其他任何機器學習任務。 之前的pytorch不支援Windows系統,但在最近的更新中已經支援Windows系統,單是在安裝pytorch總是出錯,許多部落格多有安裝方法,單更多的是談到了具體的安裝步驟,並未對安裝過程中的一些問題作出解答。 本人在安裝pytorch的過程中,花費了大量的時間,也閱讀了許多的部落格,多未能找到解決方法,故寫此部落格分享心得。

pytorch安裝有兩種方法:

  • 一、採用Anaconda安裝
  • Anaconda是一個很好的python資源包管理器,這種方法在許多部落格上都有提及,以此就不在重複。

二、採用pip安裝 這時一種非常簡單的方法,具體如下:

  1. 安裝pip 一般安裝python後都有

  2. pytorch安裝 到pytorch官網https://pytorch.org/get-started/locally/,選擇適合你電腦的配置:例如我的電腦Windows系統,安裝了python3.7,支援CUDA9.0 開啟cmd,鍵入 pip3 install http://download.pytorch.org/whl/cu90/torch-0.4.1-cp37-cp37m-win_amd64.whl pip3 install torchvision兩條命令就可以自動安裝,直至結束。

安裝中的問題: 採用 Anaconda有可能出現環境載入失敗和採用pip安裝出現版本不支援平臺的問題,這主要是你安裝的python是32位的,而官網提供的是64位的包,因此出現了不相容,只要將python重灌為64位就可以正確安裝了。