新手在配置pytorch過程中總會或多或少遇到些問題，同時網(wǎng)上關(guān)于pytorch的環(huán)境配置琳瑯滿目，不知道應(yīng)該按照哪個配置，這里筆者記錄一下自己在windows10下配置Pytorch的全過程。

筆者電腦環(huán)境以及安裝版本為：Windows10 企業(yè)版 + python 3.7.9 + Anaconda3 + GTX 1050顯卡 + pytorch 1.7.1

一、Anaconda安裝

這里推薦大家安裝Anaconda（但不是必須）。因為有些時候大家編程的python版本可能不同，所以一般大型項目建議一個項目使用一個python interpreter（編輯器），方便管理，而Anaconda就很方便的幫助我們做這件事情。

安裝過程非常簡單，在官網(wǎng)找到對應(yīng)版本安裝即可，官網(wǎng)安裝頁面如下：https://www.anaconda.com/products/individual#Downloads

開始菜單中打開Anaconda中的Anaconda Prompt，出現(xiàn)（base），即安裝成功，如下：

Anaconda可以很方便的管理我們我們的所有運行環(huán)境以及環(huán)境中的各種依賴包，主頁面如下：

二、安裝pytorch環(huán)境

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

輸入

conda create -n environment_name python=3.7

因為我們之前說過，我們需要給每個項目配置一個python編輯器，所以environment_name（改成你自己指定的環(huán)境名，比如就叫environment_test）就是我們配置的這個項目對應(yīng)的編譯環(huán)境的名字，里面放入我們指定的python編輯器（這里我以python 3.7為例）。

2. 出現(xiàn)上述頁面

輸入y，表示確認(rèn)安裝上述所需要的包。

出現(xiàn)上圖，環(huán)境中的python編輯器安裝完成。

當(dāng)然現(xiàn)在我們還處于（base）環(huán)境，也就是Anaconda自帶的基礎(chǔ)環(huán)境，我們需要跳到我們剛剛創(chuàng)建的environment_test環(huán)境。

3. 按照上圖

執(zhí)行conda activate environment_test，跳到environment_test編譯環(huán)境中

此時環(huán)境已經(jīng)發(fā)生改變，輸入pip list，可以看到目前環(huán)境中所有已經(jīng)安裝的包，但是沒有我們想要的pytorch，需要我們自己安裝。

4. pytorch安裝

pytorch安裝非常的人性化，我們只需要去pytorch官網(wǎng)，點擊install，就可以看見下面這個界面：

對于每一項的選擇在這里做一下詳細(xì)介紹：

• ① Pytorch Build：建議選Stable穩(wěn)定版
• ② Your OS：根據(jù)你自己系統(tǒng)選擇，這里我選Windows
• ③ Package：因為我們已經(jīng)安裝Anaconda，所以在Anaconda Prompt環(huán)境下我們可以直接使用Conda方式進(jìn)行安裝，所以選Conda（大家其實最常用發(fā)應(yīng)該是pip的方式，當(dāng)然官網(wǎng)也提供這種方式，這里就不做過多介紹了）
• ④ Language：Python
• ⑤ CUDA：關(guān)于GPU的CUDA版本問題如何選擇，打開cmd，輸入nvidia-smi，可以看到CUDA的版本信息，這里選11.0。（沒有英偉達(dá)顯卡，或者顯卡不支持CUDA或者就想用CPU運行的話直接選None就行）

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

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

注意：python 3.9 還需要在結(jié)尾增加-c=conda-forge

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

⑦ 出現(xiàn)done，安裝完成 證明pytorch安裝成功，關(guān)閉當(dāng)前Anaconda Prompt，并重新從開始菜單打開，并激活environment_test環(huán)境，在環(huán)境下打開python編輯頁面。

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

輸入torch.cuda.is_available()，顯示True，證明cuda安裝成功，可以使用GPU對pytorch中的張量、網(wǎng)絡(luò)進(jìn)行加速處理。若為False，此時pytorch只能用CPU運行。

三、可能遇到的一些問題

筆者在這里記錄第一次安裝時出現(xiàn)的一些問題，在這里記錄一下~

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

2.‘nvidia-smi’ 不是內(nèi)部或外部命令：解決方法如下

• windows下直接輸入nvidia-smi顯示不是內(nèi)部或外部命令也不是可運行的程序

• 把C:\Program Files\NVIDIA Corporation\NVSMI
• 添加到path。再重新打開cmd窗口

• ok，解決啦~

3. 環(huán)境中安裝jupyter，報react404之類的錯誤：解決方法如下

• [W xx:xx:xx.xxx NotebookApp] 404 GET/static/components/react/react-dom.production.min.js (::1)
• jupyter notebook打不開
• 問題描述
• 在不知道conda了一些奇奇怪怪的安裝包的時候，出現(xiàn)了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安裝成功，一定要關(guān)閉當(dāng)前Anaconda Prompt，再重新打開激活環(huán)境，因為新安裝完之后環(huán)境是未激活狀態(tài)，這時候一些依賴包是不能被調(diào)用的

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

① 未正確安裝GPU，查看 任務(wù)管理器–性能，下面是否有你的顯卡 

② CUDA9.2以上版本支持的Driver Version不可以低于396.26，一般建議安裝前先更新一下自己顯卡的驅(qū)動。不建議去英偉達(dá)官網(wǎng)更新，因為查詢最新的驅(qū)動有很多，不知道會安裝哪個。直接英偉達(dá)顯卡設(shè)置管理軟件中安裝即可，見上文。

③ 安裝的CUDA版本不能高于顯卡支持的CUDA版本。

以上為個人經(jīng)驗，希望能給大家一個參考，也希望大家多多支持腳本之家。

最新評論