一直以來絕大多數的深度學習應用都是基於Keras和Tensorflow搭建的,很少有去花時間使用過Pytorch,最近正好有時間就想學習下這個被稱爲最適合做學術研究的深度學習框架。
這裏主要是記錄下自己windows環境下的環境搭建過程,希望能幫到需要的人。
安裝Pytorch的方式比較特殊,雖然也是基於pip安裝的,但是又不能像安裝Keras一樣,直接pip安裝,這樣的話會報錯要你參考官網給出來的安裝方式。
先來看下Pytorch的官網,在這裏,首頁截圖如下所示:
這裏我們選擇的是本地安裝模式【Locally】,按照對應的選項點擊選擇完成之後,對應的安裝命令就已經出現在最下方的文本框裏面了,比如這裏對於我CPU的安裝方式來說,安裝Pytorch的命令如下:
pip install torch==1.5.0+cpu torchvision==0.6.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
複製到CMD窗口裏面,直接執行就好了,成功安裝後,導入使用截圖如下所示:
到此,確定安裝成功,後面可以學習使用了,記錄一下。