關於PyTorch環境配置及安裝教程(Windows10)

Posted on by

新手在配置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 Files\NVIDIA Corporation\NVSMI
  • 添加到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版本。

以上為個人經驗，希望能給大傢一個參考，也希望大傢多多支持WalkonNet。

推薦閱讀：