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

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

一、Anaconda安裝

這里推薦大家安裝Anaconda（但不是必須）。因?yàn)橛行r(shí)候大家編程的python版本可能不同，所以一般大型項(xiàng)目建議一個(gè)項(xiàng)目使用一個(gè)python interpreter（編輯器），方便管理，而Anaconda就很方便的幫助我們做這件事情。

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

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

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

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

1. 停留在上面 Anaconda Prompt 頁(yè)面上

輸入

conda create -n environment_name python=3.7

因?yàn)槲覀冎罢f(shuō)過(guò)，我們需要給每個(gè)項(xiàng)目配置一個(gè)python編輯器，所以environment_name（改成你自己指定的環(huán)境名，比如就叫environment_test）就是我們配置的這個(gè)項(xiàng)目對(duì)應(yīng)的編譯環(huán)境的名字，里面放入我們指定的python編輯器（這里我以python 3.7為例）。

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

輸入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)境中

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

4. pytorch安裝

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

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

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

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

• ⑥ 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，并重新從開(kāi)始菜單打開(kāi)，并激活environment_test環(huán)境，在環(huán)境下打開(kāi)python編輯頁(yè)面。

輸入import torch，不報(bào)錯(cuò)證明pytorch包安裝成功。

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

三、可能遇到的一些問(wèn)題

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

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

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

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

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

• ok，解決啦~

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

• [W xx:xx:xx.xxx NotebookApp] 404 GET/static/components/react/react-dom.production.min.js (::1)
• jupyter notebook打不開(kāi)
• 問(wèn)題描述
• 在不知道conda了一些奇奇怪怪的安裝包的時(shí)候，出現(xiàn)了jupyter 打不開(kāi)的錯(cuò)誤 錯(cuò)誤代碼為以下。

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

• 問(wèn)題解決方法
• 在csdn某個(gè)博主上看的文章，忘記是誰(shuí)了，cmd下輸入一下這個(gè)就好了

pip install --user --ignore-installed jupyter

• ok，解決啦~

4. 上述的第5步證明pytorch安裝成功，一定要關(guān)閉當(dāng)前Anaconda Prompt，再重新打開(kāi)激活環(huán)境，因?yàn)樾掳惭b完之后環(huán)境是未激活狀態(tài)，這時(shí)候一些依賴包是不能被調(diào)用的

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

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

② CUDA9.2以上版本支持的Driver Version不可以低于396.26，一般建議安裝前先更新一下自己顯卡的驅(qū)動(dòng)。不建議去英偉達(dá)官網(wǎng)更新，因?yàn)椴樵冏钚碌尿?qū)動(dòng)有很多，不知道會(huì)安裝哪個(gè)。直接英偉達(dá)顯卡設(shè)置管理軟件中安裝即可，見(jiàn)上文。

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

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

最新評(píng)論