win10+CPU+Python3.6下安裝pytorch
阿新 • • 發佈:2019-01-01
- 寫在前面
最近,學習之餘,想繼續搞以前的深度學習。雖然電腦上已經安裝配置好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之旅啦!