1、Pytorch

深度學(xué)習(xí)跑代碼的時(shí)候，因?yàn)楹?jiǎn)單的操作不適合cpu運(yùn)行，我們更習(xí)慣用GPU加速代碼。

本章將介紹怎么安裝pytorch的gpu環(huán)境，以及常見(jiàn)的問(wèn)題

關(guān)于conda的安裝，參考之前文章：深度學(xué)習(xí)環(huán)境配置：Anaconda 安裝和 pip 源

pytorch官網(wǎng)提供的安裝：鏈接：https://pytorch.org/

這里提供的版本都是較新的，電腦的硬件跟不上的話(huà)，可以選擇之前版本的

2、如何查看電腦的cuda版本

網(wǎng)上有很多，這里介紹自己使用的

需要注意的，只有英偉達(dá)的顯卡支持gpu加速，英特爾的不行

鼠標(biāo)右鍵有NVIDIA 控制面板這個(gè)選項(xiàng)，就代表電腦有英偉達(dá)的顯卡（window11 右鍵可能被隱藏了，點(diǎn)開(kāi)顯示更多選項(xiàng)展開(kāi)即可）

可能配置不同，打開(kāi)的界面不一樣，沒(méi)關(guān)系，繼續(xù)操作就行了

點(diǎn)開(kāi) NVIDIA控制面板

點(diǎn)這

點(diǎn)開(kāi)后，再點(diǎn)擊這里：點(diǎn)擊組件

如下：

說(shuō)明電腦支持的最大cuda版本是12.0，安裝小于等于這個(gè)版本的都可以

3、安裝cuda

這里不建議選擇最新的，選擇之前的穩(wěn)定一點(diǎn)，這里展示的是10.2版本

在conda新建好環(huán)境后，并且激活，利用pip安裝即可

換成pip安裝，將3刪除就行了

等待下載完成即可：

4、檢查是否安裝成功

首先還是要激活環(huán)境啊，如果你跟著本章來(lái)的話(huà)，已經(jīng)在環(huán)境里面了就不需要激活

首先進(jìn)入python環(huán)境，導(dǎo)入torch，然后輸入torch.cuda.is_available()

當(dāng)出現(xiàn)True的時(shí)候，那么恭喜你，gpu的配置成功了?。。?！

5、關(guān)于庫(kù)文件用conda和pip哪個(gè)安裝？

這里建議在虛擬環(huán)境中，使用pip安裝，實(shí)在安裝不來(lái)，再用conda

盡量不要來(lái)回混肴使用！！

顯卡問(wèn)題，

6、安裝 requirement 文件，配置好的gpu是否會(huì)失效？

這里之前在跑別人項(xiàng)目的時(shí)候，總會(huì)出現(xiàn)這個(gè)問(wèn)題，當(dāng)然沒(méi)人解答，大部分人都說(shuō)環(huán)境壞了，在新建一個(gè)就行了....

首先重點(diǎn)是，跑新的項(xiàng)目，一定要新建虛擬環(huán)境，這樣才不會(huì)造成破壞，俗稱(chēng)環(huán)境污染？

那么新建好了虛擬環(huán)境，我辛辛苦苦看cuda版本，又去官網(wǎng)扒了好久歷史版本才安裝好的gpu環(huán)境。怎么能保證我隨便pip install -r requirement就能保證gpu可以運(yùn)行？

下面是自己的一些理解：

需要知道是，cuda和torch版本是兩個(gè)概念。例如cuda的11版本，對(duì)應(yīng)torch好多個(gè)小版本

torch版本無(wú)非有兩種，相對(duì)于電腦的硬件來(lái)說(shuō)，要么高了，要么低了

低了的才好，因?yàn)橛布窍蛳录嫒莸模褪钦f(shuō)你cuda是11版本的。你跑的項(xiàng)目torch只要低于11版本，那么隨便怎么安裝requirement文件，都能保證gpu運(yùn)行。對(duì)于大部分成熟的項(xiàng)目來(lái)說(shuō)，torch版本都不會(huì)太高，因?yàn)樵礁咴讲环€(wěn)定，開(kāi)發(fā)者也不希望別人復(fù)現(xiàn)代碼的時(shí)候一直報(bào)紅。

第二種就是高了，高了的話(huà)，電腦肯定不支持cuda運(yùn)行了。當(dāng)然也不是不能解決

需要明白的是，不是開(kāi)發(fā)者非要炫耀自己硬件好，才選擇高的版本。有些功能在高的torch版本才有，例如混合精度計(jì)算，對(duì)于深度學(xué)習(xí)來(lái)說(shuō)已經(jīng)很常見(jiàn)了，好像torch1.10版本才支持，這種是沒(méi)辦法的。

如果真的高了，建議新建好環(huán)境的時(shí)候，看看torch的版本，如果和低版本的torch適應(yīng)的話(huà)，在自身配置符合的情況下可以低幾個(gè)小版本也沒(méi)事。大不了在requirement文件把torch刪除了就行了，咱們自己配置torch

當(dāng)然，肯定有人不知道低多少合適。舉個(gè)例子，你電腦的cuda是11版本的，別人項(xiàng)目需要12版本的cuda。看他torch的對(duì)應(yīng)版本，往下降一點(diǎn)，不就是電腦可以支持的11版本了嗎。要是還不知道，安裝自身電腦最高的torch版本，然后跑一遍看看報(bào)不報(bào)錯(cuò)就行了

總結(jié)

要是真的是配置跟不上，那真沒(méi)辦法了，搞深度學(xué)習(xí)還是需要鈔能力...

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

最新評(píng)論