那些最全面的Windows10安裝pytorch踩過的坑以及如何應用

格物兄發表於2021-01-17

那些最全面的Windows10安裝pytorch踩過的坑以及如何應用

一、pytorch簡介

2017年1月,由Facebook人工智慧研究院(FAIR)基於Torch推出了PyTorch。它是一個基於Python的可續計算包,提供兩個高階功能:1、具有強大的GPU加速的張量計算(如NumPy)。2、包含自動求導系統的深度神經網路。

深度學習的框架有很多,諸如選擇哪個深度學習的框架,哪個框架好這類的問題很多,這裡引用知乎的回答 PyTorch到底好用在哪裡? PyTorch更有利於研究人員、愛好者、小規模專案等快速搞出原型。而TensorFlow更適合大規模部署,特別是需要跨平臺和嵌入式部署時。

因此,筆者選擇在Windows10作業系統下安裝pytorch,但是確實遇到了很多坑,很多在Linux上只要幾個命令列就解決的問題,把自己遇到的問題跟網友分享下。

二、踩過的坑及詳細解決方法

本人是在Anaconda環境下安裝pytorch的,因為Anaconda確實是很好的管理環境,無腦安裝就可以了。強烈建議在Anaconda中構建虛擬環境後,安裝你的pytorch版本,因為不同版本的pytorch或者不同專案需要不同的pytorch版本差別還是有的,這樣便於進行版本的適配。介紹完安裝的環境,下面就開始具體講解安裝過程中有哪些坑?

(一)安裝顯示卡驅動

這裡主要是訪問英偉達官網,根據自己電腦顯示卡的版本,作業系統的版本,選擇適配的顯示卡驅動的版本,下載安裝:

英偉達顯示卡驅動官網地址

(二)Anaconda配置虛擬環境

步驟一:建立conda環境 conda create -n pytorch python=3.6 出現condaHTTPERROR

解決方法:這是由於根據下載路徑訪問時,下載速度太慢,導致的錯誤。解決辦法是修改下載路徑,更改為清華映象源的路徑,方法有多種,一是直接修改 C:\Users\使用者名稱.condarc檔案 ,例如:

`channels:

三、如何應用

下面主要針對jupyter notebook和pycharm兩種應用環境進行講解

(一)在jupyter notebook裡使用虛擬環境下的pytorch

步驟一:開啟終端Anaconda Prompt(管理員許可權開啟,右鍵單擊,選擇以管理員身份執行就可以)

步驟二:安裝nb_conda外掛

1.執行以下命令列:

conda install nb_conda

選擇“Y”即可

2.執行命令列:

conda install ipykernel

注意,這一步需要在你的pytorch虛擬環境下安裝,先輸入 conda activate pytorch(這裡的pytorch是我的虛擬環境,根據自己設定的名稱來),進入環境後執行conda install ipykernel.

步驟三:啟動jupyter notebook

 

(二)在pycharm裡使用虛擬環境的pytorch

步驟一:開啟pycharm,在files->settings->project:<xxx>->project interpreter介面,可以設定project interpreter

步驟二:選擇【齒輪】,點選【Add】,後進入:

步驟三:選擇【齒輪】,點選【Add】,後進入:

步驟四:選擇一個已經存在的conda虛擬環境,即【pytorch】,點選【OK】,進入上一層:

步驟五:點選【OK】,回退上一層:

這樣就能在pycharm裡面設定anaconda虛擬環境下的pytorch進行程式設計開發了。

 

碼字不易,請關注微信公眾號,不丟失

 

相關文章