基于Anaconda安裝pytorch深度學(xué)習(xí)環(huán)境+pycharm安裝---免額外安裝CUDA和cudnn

前言

最近由于項(xiàng)目需要，之前我們?cè)诶?a href="http://ttokpm.com/tags/gpu/" target="_blank">GPU進(jìn)行深度學(xué)習(xí)的時(shí)候，都要去NVIDIA的官網(wǎng)下載CUDA的安裝程序和cudnn的壓縮包，然后再進(jìn)行很繁瑣的系統(tǒng)環(huán)境配置。不僅環(huán)境配置麻煩，而且還特別容易配置錯(cuò)誤，特別還有CUDA和cudnn版本的對(duì)應(yīng)也特別容易搞錯(cuò)，但是利用anaconda安裝配置pytorch和paddle環(huán)境的時(shí)候會(huì)自動(dòng)幫我們配置好cuda和cudnn。這篇文章可以幫助小白快速配置深度學(xué)習(xí)環(huán)境。

一、NVIDIA驅(qū)動(dòng)安裝

顯卡驅(qū)動(dòng)程序就是用來(lái)驅(qū)動(dòng)顯卡的程序，它是硬件所對(duì)應(yīng)的軟件。驅(qū)動(dòng)程序即添加到操作系統(tǒng)中的一小塊代碼，其中包含有關(guān)硬件設(shè)備的信息。正常有顯卡的電腦都是有驅(qū)動(dòng)程序的，但是有的時(shí)候驅(qū)動(dòng)可能版本比較低，支持的cuda版本也是比較低的（但是有的人的顯卡是比較老的，就不建議更新驅(qū)動(dòng)，這樣會(huì)導(dǎo)致各種各樣的問(wèn)題，但是搞深度學(xué)習(xí)還是要用一塊好的顯卡用來(lái)學(xué)習(xí)，這點(diǎn)是過(guò)來(lái)人有血淚教訓(xùn)的，如果顯卡性能強(qiáng)悍，可以大大節(jié)省AI模型訓(xùn)練時(shí)間，提高效率。）

首先查看電腦的顯卡版本，步驟為：此電腦右擊-->管理-->設(shè)備管理器-->顯示適配器。就可以看到電腦顯卡的版本了。

有顯卡驅(qū)動(dòng)的，可以直接在桌面右鍵，找到英偉達(dá)驅(qū)動(dòng)控制面板打開(kāi)就好了。

在得知以上的信息以后我們就可以對(duì)應(yīng)我們的顯卡去英偉達(dá)官網(wǎng)上去找相對(duì)應(yīng)的顯卡驅(qū)動(dòng)更新或者下載了。

顯卡驅(qū)動(dòng)的下載地址（可能打開(kāi)比較慢，多打開(kāi)幾遍）。

然后根據(jù)自己的電腦的配置去選擇驅(qū)動(dòng)。這里Notebooks是筆記本的意思，所以如果你是筆記本電腦，那么產(chǎn)品系列那個(gè)選項(xiàng)就要選（Notebooks）的。還有那個(gè)下載類型有兩種一個(gè)是Studio版本，一個(gè)是Game Ready版本。其實(shí)兩個(gè)版本都差不多，一個(gè)是偏辦公用，一個(gè)是偏游戲娛樂(lè)。按如下操作將驅(qū)動(dòng)下載下來(lái)。然后點(diǎn)擊下載來(lái)的程序，不斷的下一步就好了。

安裝（更新）好了顯卡驅(qū)動(dòng)以后。我們按下win+R組合鍵，打開(kāi)cmd命令窗口。輸入如下的命令。

nvidia-smi

得到如下圖的信息圖，可以看到驅(qū)動(dòng)的版本是442.50；最高支持的CUDA版本是10.2版本。得到顯卡的最高支持的CUDA版本，我們就可以根據(jù)這個(gè)信息來(lái)安裝環(huán)境了。

二、Anaconda 的安裝

打開(kāi)網(wǎng)址，現(xiàn)在是2021年10月，對(duì)應(yīng)的anaconda版本是支持python3.8。如果想下載之前的版本，或者更低python版本的anaconda，可以打開(kāi)網(wǎng)址。 雙擊下載好的anaconda安裝包，just me是說(shuō)只供當(dāng)前用戶使用。all user 是供使用這臺(tái)電腦的所有用戶使用，是權(quán)限問(wèn)題。對(duì)空間影響不大。如果你的電腦上只有建了一個(gè)用戶，all users和just me 的作用是一樣的。所以點(diǎn)擊just me就好了。 然后點(diǎn)擊next，當(dāng)讓你選擇安裝安裝路徑的時(shí)候，一定不要選擇默認(rèn)安裝位置，因?yàn)槟J(rèn)位置是c盤(pán)，以后要在anaconda里面創(chuàng)建環(huán)境的時(shí)候會(huì)很占內(nèi)存，最好在D盤(pán)中創(chuàng)建一個(gè)文件夾來(lái)放anaconda。 和圖中一樣將圖中的√勾上，雖然出現(xiàn)紅色的警告，但是要勾上，將anaconda添加到環(huán)境變量中去。然后點(diǎn)擊完成就好了。 安裝完成以后，按下開(kāi)始鍵（win鍵）在左邊就會(huì)出現(xiàn)anaconda3這個(gè)文件夾，可以發(fā)現(xiàn)anaconda已經(jīng)安裝好了。

三、Pytorch環(huán)境安裝

按下開(kāi)始鍵（win鍵），點(diǎn)擊如圖中的圖標(biāo)。打開(kāi)anaconda的終端。

??????執(zhí)行如下的指令查看有哪些環(huán)境

condaenvlist

可以看出來(lái)，新安裝的anaconda只有一個(gè)base環(huán)境 這里先講一下anaconda環(huán)境，首先base環(huán)境是一個(gè)大的環(huán)境，類似一個(gè)很大的一個(gè)房子（但是沒(méi)有房間），當(dāng)我們每創(chuàng)建一個(gè)環(huán)境就都會(huì)相當(dāng)于在這個(gè)大房子里面用隔板創(chuàng)建一個(gè)房間，然后這個(gè)房間里面可以安裝我們所需要的包，這樣管理起來(lái)就比較方便。如圖可以比較直觀的詮釋anaconda的環(huán)境 創(chuàng)建虛擬環(huán)境conda create -n 環(huán)境名字（英文） python=x.x（python版本），如下，我就是創(chuàng)建了一個(gè)名字叫pytorch，python是3.8版本的環(huán)境。

condacreate-npytorchpython=3.8

在base環(huán)境中執(zhí)行如上的命令，就會(huì)創(chuàng)建一個(gè)新的虛擬環(huán)境，這個(gè)虛擬環(huán)境會(huì)安裝一些基礎(chǔ)的包，如下圖所示。詢問(wèn)是否安裝的時(shí)候，輸入y。就可以創(chuàng)建環(huán)境了。 當(dāng)安裝好了以后，執(zhí)行conda env list這個(gè)命令，就可以看到比一開(kāi)始多了一個(gè)pytorch這個(gè)環(huán)境?，F(xiàn)在我們可以在這個(gè)環(huán)境里面安裝深度學(xué)習(xí)框架和一些Python包了。 執(zhí)行如下命令，激活這個(gè)環(huán)境。conda activate 虛擬環(huán)境名稱

condaactivatepytorch

安裝pytorch-gup版的環(huán)境，由于pytorch的官網(wǎng)在國(guó)外，下載相關(guān)的環(huán)境包是比較慢的，所以我們給環(huán)境換源。在pytorch環(huán)境下執(zhí)行如下的命名給環(huán)境換清華源。

condaconfig--addchannelshttps://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
condaconfig--addchannelshttps://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
condaconfig--addchannelshttps://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
condaconfig--setshow_channel_urlsyes

??????然后打開(kāi)pytorch的官網(wǎng)，由于開(kāi)頭我們通過(guò)驅(qū)動(dòng)檢測(cè)到我的顯卡為 GeForce MX350，最高支持cuda10.2版本，所以我們選擇cuda10.2版本的cuda，然后將下面紅色框框中的內(nèi)容復(fù)制下來(lái)。

??????將復(fù)制的內(nèi)容粘貼到pytorch環(huán)境下的終端，運(yùn)行就可以了

condainstallpytorch==1.12.1torchvision==0.13.1torchaudio==0.12.1cudatoolkit=10.2-cpytorch

這時(shí)候就開(kāi)始下載環(huán)境所需要的依賴包了。

四、pycharm安裝--驗(yàn)證CUDA和cudnn版本

打開(kāi)這個(gè)pycharm網(wǎng)址，可以發(fā)現(xiàn)一共有兩個(gè)版本一個(gè)是專業(yè)版（Professional），一個(gè)是社區(qū)版（Community），專業(yè)版是需要花錢(qián)的，好幾百美元一年。而社區(qū)版是免費(fèi)的，但是也夠用了，所以就下載安裝社區(qū)版就好了。 ? ? ??將下載好的pycharm安裝包，雙擊。安裝的地址最好不要放在C盤(pán)，我是安裝在D盤(pán)（反正我 的軟件是不喜歡安裝在C盤(pán)的?？偤ε翪盤(pán)會(huì)滿）。 ? ? ??將所有的√都勾上， 安裝好了點(diǎn)擊第二個(gè)框框，然后點(diǎn)完成就好了 打開(kāi)pycharm，創(chuàng)建一個(gè)新的工程，來(lái)測(cè)試anaconda是否在安裝pytorch和paddlepaddle框架的時(shí)候也安裝了cuda和cudnn。按如下兩圖創(chuàng)建一個(gè)工程（新工程好像必須要安裝一個(gè)新的python插件），新的工程最好在D盤(pán)一個(gè)新的文件夾下，有的工程很大，C盤(pán)容易裝滿。 按以上的方式創(chuàng)建了一個(gè)工程，這時(shí)候我們就要選擇我們?cè)赼naconda里面安裝的環(huán)境，在界面的右下角 按照如圖中的選項(xiàng)選擇我們?cè)赼naconda中創(chuàng)建的深度學(xué)習(xí)環(huán)境，可以看到有pytorch環(huán)境。

此時(shí)剛剛的右下角已經(jīng)有了我們剛剛選擇的pytorch環(huán)境中的python了。 ? ? ?我們?cè)趧?chuàng)建的工程里面創(chuàng)建一個(gè)python腳本，在腳本中運(yùn)行如下代碼，查看是否anconda在安裝pytorch環(huán)境的時(shí)候也安裝了cuda和cudnn。

importtorch
print(torch.cuda.is_available())
print(torch.backends.cudnn.is_available())
print(torch.cuda_version)
print(torch.backends.cudnn.version())

可以發(fā)現(xiàn)控制臺(tái)打印出兩個(gè)True，可以說(shuō)明cuda和cudnn已經(jīng)安裝。并且可以得到cuda的版本為10.2和cudnn的版本為7.6.5版本。

至此我們的深度學(xué)習(xí)環(huán)境安裝就已經(jīng)完全完成。接下來(lái)可以在相應(yīng)的環(huán)境下進(jìn)行深度學(xué)習(xí)的實(shí)驗(yàn)了。

最后還要申明一下，你可以創(chuàng)建不同的環(huán)境，在里面安裝不同版本的cuda和cudnn版本。已經(jīng)親自嘗試過(guò)了，是可以的。