Pytorch配置GPU環(huán)境方式
1、Pytorch
深度學(xué)習(xí)跑代碼的時(shí)候，因?yàn)楹?jiǎn)單的操作不適合cpu運(yùn)行，我們更習(xí)慣用GPU加速代碼。
本章將介紹怎么安裝pytorch的gpu環(huán)境，以及常見的問題
關(guān)于conda的安裝，參考之前文章：深度學(xué)習(xí)環(huán)境配置：Anaconda 安裝和 pip 源
pytorch官網(wǎng)提供的安裝：鏈接：https://pytorch.org/
這里提供的版本都是較新的，電腦的硬件跟不上的話，可以選擇之前版本的
2、如何查看電腦的cuda版本
網(wǎng)上有很多，這里介紹自己使用的
需要注意的，只有英偉達(dá)的顯卡支持gpu加速，英特爾的不行
鼠標(biāo)右鍵有NVIDIA 控制面板這個(gè)選項(xiàng)，就代表電腦有英偉達(dá)的顯卡（window11 右鍵可能被隱藏了，點(diǎn)開顯示更多選項(xiàng)展開即可）
可能配置不同，打開的界面不一樣，沒關(guān)系，繼續(xù)操作就行了
點(diǎn)開 NVIDIA控制面板
點(diǎn)這
點(diǎn)開后，再點(diǎn)擊這里：點(diǎn)擊組件
如下：
說明電腦支持的最大cuda版本是12.0，安裝小于等于這個(gè)版本的都可以
3、安裝cuda
這里不建議選擇最新的，選擇之前的穩(wěn)定一點(diǎn)，這里展示的是10.2版本
在conda新建好環(huán)境后，并且激活，利用pip安裝即可
換成pip安裝，將3刪除就行了
等待下載完成即可：
4、檢查是否安裝成功
首先還是要激活環(huán)境啊，如果你跟著本章來的話，已經(jīng)在環(huán)境里面了就不需要激活
首先進(jìn)入python環(huán)境，導(dǎo)入torch，然后輸入torch.cuda.is_available()
當(dāng)出現(xiàn)True的時(shí)候，那么恭喜你，gpu的配置成功了?。。。?/p>
5、關(guān)于庫(kù)文件用conda和pip哪個(gè)安裝？
這里建議在虛擬環(huán)境中，使用pip安裝，實(shí)在安裝不來，再用conda
盡量不要來回混肴使用??！
顯卡問題，
6、安裝 requirement 文件，配置好的gpu是否會(huì)失效？
這里之前在跑別人項(xiàng)目的時(shí)候，總會(huì)出現(xiàn)這個(gè)問題，當(dāng)然沒人解答，大部分人都說環(huán)境壞了，在新建一個(gè)就行了....
首先重點(diǎn)是，跑新的項(xiàng)目，一定要新建虛擬環(huán)境，這樣才不會(huì)造成破壞，俗稱環(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版本無非有兩種，相對(duì)于電腦的硬件來說，要么高了，要么低了
低了的才好，因?yàn)橛布窍蛳录嫒莸模褪钦f你cuda是11版本的。你跑的項(xiàng)目torch只要低于11版本，那么隨便怎么安裝requirement文件，都能保證gpu運(yùn)行。對(duì)于大部分成熟的項(xiàng)目來說，torch版本都不會(huì)太高，因?yàn)樵礁咴讲环€(wěn)定，開發(fā)者也不希望別人復(fù)現(xiàn)代碼的時(shí)候一直報(bào)紅。
第二種就是高了，高了的話，電腦肯定不支持cuda運(yùn)行了。當(dāng)然也不是不能解決
需要明白的是，不是開發(fā)者非要炫耀自己硬件好，才選擇高的版本。有些功能在高的torch版本才有，例如混合精度計(jì)算，對(duì)于深度學(xué)習(xí)來說已經(jīng)很常見了，好像torch1.10版本才支持，這種是沒辦法的。
如果真的高了，建議新建好環(huán)境的時(shí)候，看看torch的版本，如果和低版本的torch適應(yīng)的話，在自身配置符合的情況下可以低幾個(gè)小版本也沒事。大不了在requirement文件把torch刪除了就行了，咱們自己配置torch
當(dāng)然，肯定有人不知道低多少合適。舉個(gè)例子，你電腦的cuda是11版本的，別人項(xiàng)目需要12版本的cuda?？此鹴orch的對(duì)應(yīng)版本，往下降一點(diǎn)，不就是電腦可以支持的11版本了嗎。要是還不知道，安裝自身電腦最高的torch版本，然后跑一遍看看報(bào)不報(bào)錯(cuò)就行了
總結(jié)
要是真的是配置跟不上，那真沒辦法了，搞深度學(xué)習(xí)還是需要鈔能力...
以上為個(gè)人經(jīng)驗(yàn)，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
python3 使用OpenCV計(jì)算滑塊拼圖驗(yàn)證碼缺口位置(場(chǎng)景示例)
這篇文章主要介紹了python3 使用OpenCV計(jì)算滑塊拼圖驗(yàn)證碼缺口位置,本文通過場(chǎng)景示例給大家詳細(xì)介紹，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友可以參考下2021-08-08
chardet?是一個(gè)流行的 Python 庫(kù),用于檢測(cè)文本文件的字符編碼,本文就來介紹一下python基于chardet識(shí)別字符編碼的方法,具有一定的參考價(jià)值,感興趣的可以了解一下2025-01-01
如何將Yolov5的detect.py修改為可以直接調(diào)用的函數(shù)詳解
YOLOv4還沒有退熱,YOLOv5已經(jīng)發(fā)布,下面這篇文章主要給大家介紹了關(guān)于如何將Yolov5的detect.py修改為可以直接調(diào)用的函數(shù)的相關(guān)資料,文中通過示例代碼介紹的非常詳細(xì),需要的朋友可以參考下2022-04-04
這篇文章主要介紹了Python的SQLAlchemy框架使用入門,SQLAlchemy框架是Python中用來操作數(shù)據(jù)庫(kù)的ORM框架之一,需要的朋友可以參考下2015-04-04
Python常用標(biāo)準(zhǔn)庫(kù)之os模塊功能
這篇文章主要介紹了Python常用標(biāo)準(zhǔn)庫(kù)之os模塊功能,os模塊的主要功能有系統(tǒng)相關(guān)、目錄及文件操作、執(zhí)行命令和管理進(jìn)程，其中的進(jìn)程管理功能主要是Linux相關(guān)的，此處不做討論，對(duì)Python標(biāo)準(zhǔn)庫(kù)os相關(guān)知識(shí)感興趣的朋友跟隨小編一起看看吧2022-11-11
Python的五個(gè)標(biāo)準(zhǔn)數(shù)據(jù)類型你認(rèn)識(shí)幾個(gè)
這篇文章主要為大家詳細(xì)介紹了Python標(biāo)準(zhǔn)數(shù)據(jù)類型，文中示例代碼介紹的非常詳細(xì)，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下，希望能夠給你帶來幫助2022-03-03
使用Python創(chuàng)建多功能文件管理器的代碼示例
在本文中,我們將探索一個(gè)使用Python的wxPython庫(kù)開發(fā)的文件管理器應(yīng)用程序,這個(gè)應(yīng)用程序不僅能夠?yàn)g覽和選擇文件,還支持文件預(yù)覽、壓縮、圖片轉(zhuǎn)換以及生成PPT演示文稿的功能,需要的朋友可以參考下2024-08-08
這篇文章主要介紹了Python中集合的內(nèi)置函數(shù)詳解,集合在python里面是另外一種對(duì)象,它的特征是在大括號(hào)里面,并且各個(gè)元素是唯一的,今天我們來看一下其內(nèi)置了哪些函數(shù),需要的朋友可以參考下2023-08-08
Python實(shí)現(xiàn)返回?cái)?shù)組中第i小元素的方法示例
這篇文章主要介紹了Python實(shí)現(xiàn)返回?cái)?shù)組中第i小元素的方法,結(jié)合實(shí)例形式分析了Python針對(duì)數(shù)組的遍歷、排序、運(yùn)算等相關(guān)操作技巧,需要的朋友可以參考下2017-12-12
淺談PyQt5中異步刷新UI和Python多線程總結(jié)
今天小編就為大家分享一篇淺談PyQt5中異步刷新UI和Python多線程總結(jié)，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2019-12-12