快捷導(dǎo)航

關(guān)于PyTorch環(huán)境配置及安裝教程(Windows10)

更新時(shí)間：2022年07月31日 11:02:12 作者：酷酷的橙007

這篇文章主要介紹了關(guān)于PyTorch環(huán)境配置及安裝教程(Windows10)，具有很好的參考價(jià)值，希望對大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教

一、Anaconda安裝

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

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

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

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

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

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

輸入

conda create -n environment_name python=3.7

因?yàn)槲覀冎罢f過，我們需要給每個(gè)項(xiàng)目配置一個(gè)python編輯器，所以environment_name（改成你自己指定的環(huán)境名，比如就叫environment_test）就是我們配置的這個(gè)項(xiàng)目對應(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)境中

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

4. pytorch安裝

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

對于每一項(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)也提供這種方式，這里就不做過多介紹了）
④ Language：Python
⑤ CUDA：關(guān)于GPU的CUDA版本問題如何選擇，打開cmd，輸入nvidia-smi，可以看到CUDA的版本信息，這里選11.0。（沒有英偉達(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，并重新從開始菜單打開，并激活environment_test環(huán)境，在環(huán)境下打開python編輯頁面。

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

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

三、可能遇到的一些問題

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

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

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

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

把C:\Program Files\NVIDIA Corporation\NVSMI
添加到path。再重新打開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打不開
問題描述
在不知道conda了一些奇奇怪怪的安裝包的時(shí)候，出現(xiàn)了jupyter 打不開的錯(cuò)誤錯(cuò)誤代碼為以下。

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

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

pip install --user --ignore-installed jupyter

ok，解決啦~

4. 上述的第5步證明pytorch安裝成功，一定要關(guān)閉當(dāng)前Anaconda Prompt，再重新打開激活環(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è)置管理軟件中安裝即可，見上文。

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

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

您可能感興趣的文章: