首頁 > 軟體

關於PyTorch環境設定及安裝教學(Windows10)

2022-07-31 14:00:26

新手在設定pytorch過程中總會或多或少遇到些問題,同時網上關於pytorch的環境設定琳琅滿目,不知道應該按照哪個設定,這裡筆者記錄一下自己在windows10下設定Pytorch的全過程。

筆者電腦環境以及安裝版本為:Windows10 企業版 + python 3.7.9 + Anaconda3 + GTX 1050顯示卡 + pytorch 1.7.1

一、Anaconda安裝

這裡推薦大家安裝Anaconda(但不是必須)。因為有些時候大家程式設計的python版本可能不同,所以一般大型專案建議一個專案使用一個python interpreter(編輯器),方便管理,而Anaconda就很方便的幫助我們做這件事情。

安裝過程非常簡單,在官網找到對應版本安裝即可,官網安裝頁面如下:https://www.anaconda.com/products/individual#Downloads

開始選單中開啟Anaconda中的Anaconda Prompt,出現(base),即安裝成功,如下:

Anaconda可以很方便的管理我們我們的所有執行環境以及環境中的各種依賴包,主頁面如下:

二、安裝pytorch環境

1. 停留在上面 Anaconda Prompt 頁面上

輸入

conda create -n environment_name python=3.7

因為我們之前說過,我們需要給每個專案設定一個python編輯器,所以environment_name(改成你自己指定的環境名,比如就叫environment_test)就是我們設定的這個專案對應的編譯環境的名字,裡面放入我們指定的python編輯器(這裡我以python 3.7為例)。

2. 出現上述頁面

輸入y,表示確認安裝上述所需要的包。

出現上圖,環境中的python編輯器安裝完成。

當然現在我們還處於(base)環境,也就是Anaconda自帶的基礎環境,我們需要跳到我們剛剛建立的environment_test環境。

3. 按照上圖

執行conda activate environment_test,跳到environment_test編譯環境中

此時環境已經發生改變,輸入pip list,可以看到目前環境中所有已經安裝的包,但是沒有我們想要的pytorch,需要我們自己安裝。

4. pytorch安裝

pytorch安裝非常的人性化,我們只需要去pytorch官網,點選install,就可以看見下面這個介面:

對於每一項的選擇在這裡做一下詳細介紹:

  • ① Pytorch Build:建議選Stable穩定版
  • ② Your OS:根據你自己系統選擇,這裡我選Windows
  • ③ Package:因為我們已經安裝Anaconda,所以在Anaconda Prompt環境下我們可以直接使用Conda方式進行安裝,所以選Conda(大家其實最常用發應該是pip的方式,當然官網也提供這種方式,這裡就不做過多介紹了)
  • ④ Language:Python
  • ⑤ CUDA:關於GPU的CUDA版本問題如何選擇,開啟cmd,輸入nvidia-smi,可以看到CUDA的版本資訊,這裡選11.0。(沒有英偉達顯示卡,或者顯示卡不支援CUDA或者就想用CPU執行的話直接選None就行)

注意CUDA9.2以上版本支援的Driver Version不可以低於396.26,一般建議安裝前先更新一下自己顯示卡的驅動,筆者昨天剛剛更新完,直接點更新,很方便。

  • ⑥ Run this Command:conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch。

注意:python 3.9 還需要在結尾增加-c=conda-forge

可以看到裡面包括pytorch,numpy等我們所需要的一些包,還是輸入y,等待安裝。

⑦ 出現done,安裝完成 證明pytorch安裝成功,關閉當前Anaconda Prompt,並重新從開始選單開啟,並啟用environment_test環境,在環境下開啟python編輯頁面。

輸入import torch,不報錯證明pytorch包安裝成功。

輸入torch.cuda.is_available(),顯示True,證明cuda安裝成功,可以使用GPU對pytorch中的張量、網路進行加速處理。若為False,此時pytorch只能用CPU執行。

三、可能遇到的一些問題

筆者在這裡記錄第一次安裝時出現的一些問題,在這裡記錄一下~

1. pytorch包安裝不成功的話,建議按照提示更新一下conda版本(可能是你的Anaconda版本太老了,新安裝的應該不會出現這個問題)

2.‘nvidia-smi’ 不是內部或外部命令:解決方法如下

  • windows下直接輸入nvidia-smi顯示不是內部或外部命令也不是可執行的程式

  • 把C:Program FilesNVIDIA CorporationNVSMI
  • 新增到path。再重新開啟cmd視窗

  • ok,解決啦~

3. 環境中安裝jupyter,報react404之類的錯誤:解決方法如下

  • [W xx:xx:xx.xxx NotebookApp] 404 GET/static/components/react/react-dom.production.min.js (::1)
  • jupyter notebook打不開
  • 問題描述
  • 在不知道conda了一些奇奇怪怪的安裝包的時候,出現了jupyter 打不開的錯誤 錯誤程式碼為以下。

[W xx:xx:xx.xxx NotebookApp] 404 GET/static/components/react/react-dom.production.min.js (::1)

  • 問題解決方法
  • 在csdn某個博主上看的文章,忘記是誰了,cmd下輸入一下這個就好了
pip install --user --ignore-installed jupyter
  • ok,解決啦~

4. 上述的第5步證明pytorch安裝成功,一定要關閉當前Anaconda Prompt,再重新開啟啟用環境,因為新安裝完之後環境是未啟用狀態,這時候一些依賴包是不能被呼叫的

5. torch.cuda.is_available()顯示false的幾個原因:

① 未正確安裝GPU,檢視 工作管理員–效能,下面是否有你的顯示卡 

② CUDA9.2以上版本支援的Driver Version不可以低於396.26,一般建議安裝前先更新一下自己顯示卡的驅動。不建議去英偉達官網更新,因為查詢最新的驅動有很多,不知道會安裝哪個。直接英偉達顯示卡設定管理軟體中安裝即可,見上文。

③ 安裝的CUDA版本不能高於顯示卡支援的CUDA版本。

以上為個人經驗,希望能給大家一個參考,也希望大家多多支援it145.com。


IT145.com E-mail:sddin#qq.com