TensorFlow如何指定GPU訓(xùn)練模型
如何指定GPU訓(xùn)練模型
Linux 查看當(dāng)前服務(wù)器 GPU 的占用情況可以使用 nvidia-smi 命令，如下所示：
nvidia-smi
關(guān)于 nvidia-smi 命令輸出的詳細(xì)解釋?zhuān)蓞⒖脊P者的另外一篇文章：GPU狀態(tài)監(jiān)測(cè) nvidia-smi 命令詳解。
在此不再贅述，本文主要分享一下在用 TensorFlow 訓(xùn)練模型時(shí)如何指定 GPU。
在用 TensorFlow 訓(xùn)練深度學(xué)習(xí)模型的時(shí)候，若無(wú)手動(dòng)指定，默認(rèn)是選用第0塊來(lái)訓(xùn)練，而且其他幾塊 GPU 也會(huì)被顯示占用。
有時(shí)候，我們更希望可以自己指定一塊或者多塊 GPU 來(lái)訓(xùn)練模型，接下來(lái)介紹一種常用的也是比較有效的方法：
舉個(gè)例子
如果要指定只用第0塊 GPU 訓(xùn)練，可以在 python 代碼中如下指定：
import os os.environ['CUDA_VISIBLE_DEVICES']='0'
運(yùn)行 python 程序后，可發(fā)現(xiàn)就只有第0塊 GPU 顯示被占用，如上圖所示。
如果要指定多塊 GPU
可以如下：
os.environ['CUDA_VISIBLE_DEVICES']='1,2'
當(dāng)然，上述這種方法不太方便，每次運(yùn)行若換用 GPU 訓(xùn)練都需要改代碼，可以在運(yùn)行 python 程序的時(shí)候進(jìn)行指定：
CUDA_VISIBLE_DEVICES=0 python textCnn.py
以上為個(gè)人經(jīng)驗(yàn)，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
Python實(shí)現(xiàn)提取谷歌音樂(lè)搜索結(jié)果的方法
這篇文章主要介紹了Python實(shí)現(xiàn)提取谷歌音樂(lè)搜索結(jié)果的方法,涉及Python針對(duì)谷歌音樂(lè)相關(guān)信息的獲取技巧,具有一定參考借鑒價(jià)值,需要的朋友可以參考下2015-07-07
django跳轉(zhuǎn)頁(yè)面?zhèn)鲄⒌膶?shí)現(xiàn)
這篇文章主要介紹了django跳轉(zhuǎn)頁(yè)面?zhèn)鲄⒌膶?shí)現(xiàn)，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2020-09-09
使用Python實(shí)現(xiàn)廣告點(diǎn)擊率預(yù)測(cè)
廣告點(diǎn)擊率是指有多少用戶(hù)點(diǎn)擊了您的廣告與有多少用戶(hù)查看了您的廣告的比率,本文主要為大家介紹了如何使用Python實(shí)現(xiàn)廣告點(diǎn)擊率預(yù)測(cè),感興趣的小伙伴可以了解下2023-10-10
Python實(shí)現(xiàn)自定義讀寫(xiě)分離代碼實(shí)例
這篇文章主要介紹了Python實(shí)現(xiàn)自定義讀寫(xiě)分離代碼實(shí)例,文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2019-11-11
Python-split()函數(shù)實(shí)例用法講解
在本篇文章里小編給大家整理的是一篇關(guān)于Python-split()函數(shù)實(shí)例用法講解，有興趣的朋友們可以學(xué)習(xí)下。2020-12-12
Python數(shù)據(jù)處理利器Slice函數(shù)用法詳解
這篇文章主要給大家介紹了關(guān)于Python數(shù)據(jù)處理利器Slice函數(shù)用法的相關(guān)資料,slice函數(shù)是Python中的一個(gè)內(nèi)置函數(shù),用于對(duì)序列進(jìn)行切片操作,文中通過(guò)代碼介紹的非常詳細(xì),需要的朋友可以參考下2024-03-03
Python實(shí)現(xiàn)的對(duì)本地host127.0.0.1主機(jī)進(jìn)行掃描端口功能示例
這篇文章主要介紹了Python實(shí)現(xiàn)的對(duì)本地host127.0.0.1主機(jī)進(jìn)行掃描端口功能,可實(shí)現(xiàn)掃描本機(jī)開(kāi)放端口的功能,涉及Python socket模塊與Thread多線(xiàn)程模塊相關(guān)使用技巧,需要的朋友可以參考下2019-02-02
值得收藏，Python 開(kāi)發(fā)中的高級(jí)技巧
這篇文章主要介紹了Python 開(kāi)發(fā)中的高級(jí)技巧，非常不錯(cuò)，具有收藏價(jià)值，感興趣的朋友一起看看吧2018-11-11
本文主要介紹了numpy稀疏矩陣的實(shí)現(xiàn)，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2022-08-08
層次聚類(lèi)( Hierarchical Clustering )是聚類(lèi)算法的一種,通過(guò)計(jì)算不同類(lèi)別的相似度類(lèi)創(chuàng)建一個(gè)有層次的嵌套的樹(shù),下面這篇文章主要給大家介紹了關(guān)于Python中層次聚類(lèi)的詳細(xì)講解,需要的朋友可以參考下2022-12-12