1. 程式人生 > >win10+CPU+Python3.6下安裝pytorch

win10+CPU+Python3.6下安裝pytorch

- 寫在前面

  最近,學習之餘,想繼續搞以前的深度學習。雖然電腦上已經安裝配置好tensorflow,但是鑑於其學習難度較高,且我是一個忠實的Pythoner(愛所有Python化的東西),因此果斷入坑!
  這裡有幾篇對比tensorflow和pytorch的文章,希望能增加對二者認識。
  1:PyTorch vs TensorFlow,哪個更適合你
  2:用PyTorch還是TensorFlow?斯坦福大學CS博士生帶來全面解答
  3:PyTorch 到底好用在哪裡?

- 安裝過程

  然而,安裝過程並非一帆風順,幾度想放棄,不過最終還是堅持下來。現在看來,其實很簡單嘛。但對於很多初次接觸的人來說,總會遇到一些問題。因此,我簡單寫一下自己走過的坑。

1.支援什麼作業系統?

  現在pytorch支援Linux、MacOS、Window作業系統。其中,Window系統是18年才開始支援的,對於我這種學生黨(基本都是window)來說,無疑是個好訊息(筆者為Win10)

2. 顯示卡NV or HD

  當我網上尋找安裝教程時,發現好多都是GPU、CUDA(CUDNN),而且還要求是英偉達的顯示卡(NV),而查詢我的電腦顯示卡為AMD產的HD系列。很是苦惱。好在發現windows10 64位+cpu(無GPU) +Anaconda 安裝 pytorch這篇部落格提及了電腦非英偉達顯示卡使用者可以安裝pytorch。

3. 安裝成功

  首先登陸pytorch官網。如下:

  選擇適合你開發環境要求的選項。筆者從上到下依次選的是window、pip、3.6、None。
  檢視安裝命令

pip3 install http://download.pytorch.org/whl/cpu/torch-0.4.1-cp36-cp36m-win_amd64.whl 
pip3 install torchvision

  值得注意的是筆者這裡會報錯,需要將pip3改成pip。
  幾分鐘之後,pytorch安裝結束。開啟Spyder互動介面,引用pytorch。




  沒有報錯,表示安裝成功。現在可以開始pytorch之旅啦!