快捷導(dǎo)航

Windows下實(shí)現(xiàn)pytorch環(huán)境搭建

更新時(shí)間：2021年04月06日 08:31:11 作者：芒芒芒芒芒芒果

這篇文章主要介紹了Windows下實(shí)現(xiàn)pytorch環(huán)境搭建，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧

Anaconda安裝

安裝步驟：

1、官網(wǎng)下載安裝包：https://www.anaconda.com/distribution/

2、運(yùn)行并選擇安裝路徑，等待安裝完成。(要記得勾選 Add Anaconda to the system PATH environment variable，是為了將Anaconda添加到環(huán)境變量中)

在這里插入圖片描述

3、查看是否安裝成功，cmd中輸入conda回車(chē)，是否出現(xiàn)如下信息，有則說(shuō)明安裝成功。

在這里插入圖片描述

CUDA與CuDNN安裝

一、先檢查自己的電腦所支持的CUDA版本是多少。

步驟：

1、桌面右鍵點(diǎn)擊進(jìn)去NVIDIA控制面板，找到左下角的系統(tǒng)信息，點(diǎn)擊組件，出現(xiàn)如下界面。**

從NVCUDA.DLL 這一行后面的CUDA 10.2說(shuō)明我的電腦所支持的最高版本是10.2。

2、官網(wǎng)下載相對(duì)應(yīng)的CUDA: (https://developer.nvidia.com/cuda-toolkit-archive)

在這里插入圖片描述

我所支持的版本是10.2，所以我下載的是紅色箭頭所標(biāo)出的那行。點(diǎn)擊以后出現(xiàn)如下頁(yè)面，選擇第一個(gè)下載即可。

在這里插入圖片描述

下載完成后，在所在的文件夾下運(yùn)行安裝即可。
然后查看CUDA是否安裝成功：
cmd中運(yùn)行到安裝的文件目錄下：

cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin

然后執(zhí)行nvcc -V，查看是否出現(xiàn)如下信息，有則說(shuō)明CUDA安裝成功。

在這里插入圖片描述

如果輸入nvcc -V 出現(xiàn)錯(cuò)誤，考慮是否將Anaconds加入環(huán)境變量PATH中。

在這里插入圖片描述

這是我在實(shí)現(xiàn)中出現(xiàn)問(wèn)題的時(shí)，所解決的方法，大家如果還是不行可以自己再找找方法。

3、安裝相對(duì)應(yīng)版本的cuDNN。

（1）進(jìn)入官網(wǎng)：
https://developer.nvidia.com/rdp/cudnn-download
需要注冊(cè)并登錄賬號(hào)（這個(gè)需要一段時(shí)間），然后選擇相對(duì)應(yīng)的版本下載。

在這里插入圖片描述

我的是CUDA10.2，所以我選擇的是第二個(gè)。等待下載完成后進(jìn)行解壓，得到一個(gè)cuda文件夾，進(jìn)入之后，全選，復(fù)制到之前CUDA所安裝的文件夾下，有重復(fù)的進(jìn)行替換即可。

在這里插入圖片描述

查看cuDNN是否安裝成功：

步驟如下：（進(jìn)入安裝的路徑）

在這里插入圖片描述

出現(xiàn)如下Result = PASS 說(shuō)明cuDNN安裝成功。

在這里插入圖片描述

可以再接著執(zhí)行deviceQuery.exe，如果出現(xiàn)Result = PASS 說(shuō)明CUDA和cuDNN都已經(jīng)安裝成功了。

Pytorch安裝

1、先配置torch環(huán)境。

cmd中輸入：

conda create –n new python=3.7.4

(這兒python=xxx，大家可以先通過(guò)在cmd中直接輸入python，就會(huì)出現(xiàn)你的python的版本，這兒的new是你所創(chuàng)建的虛擬環(huán)境的名稱(chēng)，大家可以自己取)
然后激活環(huán)境：
cmd中輸入:

activate new

2、pytorch的安裝

法一：打開(kāi)pytorch官網(wǎng)：
https://pytorch.org/get-started/previous-versions/
找到自己所對(duì)應(yīng)版本的pip即可（會(huì)慢些）。

法二：進(jìn)入如下網(wǎng)址：https://download.pytorch.org/whl/torch_stable.html，下載自己所需要的torch和torchvision。
可以按快捷鍵ctrl+F進(jìn)行搜索。

在這里插入圖片描述

（cu102表示的是cuda10.2，大家根據(jù)自己cuda的版本進(jìn)行搜索，cp37是python的版本，我的是3.7.4，win就是windows系統(tǒng)了，選擇最新的版本下載即可）
查找torchvision也是同理。下載完成后，
要先進(jìn)入剛才創(chuàng)建的torch環(huán)境中，即輸入 activate new。
前面的(bash)就會(huì)變成 (new)。
接下來(lái)操作如下：

在這里插入圖片描述