PyTorch在Windows環境搭建的方法步驟
一、安裝Anaconda 3.5
Anaconda是一個用於科學計算的Python發行版,支援Linux、Mac和Window系統,提供了包管理與環境管理的功能,可以很方便地解決Python並存、切換,以及各種第三方包安裝的問題。
二、下載和安裝
個人建議推薦在清華的映象來下載。選擇合適你的版本下載,我使用的是Anaoonda3-5.1.0-Windows-x86_64.exe
可能安裝速度有點慢,不太清楚是我電腦系統盤快慢的原因還是什麼。
環境變數配置
將D:\ProgramData\Anaconda3 D:\ProgramData\Anaconda3\Library\bin D:\ProgramData\Anaconda3\Scripts 這三個配置到電腦的系統環境變數中,前面的檔案按照你安裝的檔名。
配置完了環境變數後開啟Anaconda Navigator,介面如下:
三、安裝PyTorch&torchvision
最簡單的是使用的pip命令獲取,通過進入PyTorch官網,,然後查詢自己電腦的python版本進行安裝。我是安裝的CPU版本的,當然你也可以選擇安裝CUDA,通過GPU來跑程式,GPU的確比CPU跑的快一些,因為我剛入門所以選擇CPU。
然後將下面的安裝命令在cmd中安裝即可。
pip3 install torch==1.2.0+cpu torchvision==0.4.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
接著在pycharm中執行:[/code]
這樣就表明你的torch安裝成功了!
還有一種安裝方式是conda,原理和pip其實相似,在Anaconda中Jupyter新建python檔案,執行demo
重點:如果你只想在windows上使用,使用pycharm的話只需看上面的pip安裝命令即可,一行程式碼解決問題
到此這篇關於PyTorch在Windows環境搭建的方法步驟的文章就介紹到這了,更多相關PyTorch環境搭建內容請搜尋我們以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援我們!