pytorch中多GPU并行設(shè)置

當使用GPU時蜻展,程序默認從device[0]開始運行澳化,但是如果遇到第一塊就被占滿的情況,那么程序直接報錯GPU溢出源梭,但其實第二塊第三塊空閑娱俺,因此可以通過設(shè)置GPU的Index,要求程序在哪幾塊GPU上運行。有兩種方法:
一废麻、
設(shè)置CUDA_VISIBLE_DEVICES=0,1 python train.py荠卷,指定在0,1塊上運行,

二烛愧、
在程序中使用:

import os;
os.environ['CUDA_VISIBLE_DEVICES’]=“2”油宜;

另外pytorch2.0后增加了分布式GPU運行.
分布式是指多個GPU在多臺服務器上運行,而并行是指一臺服務器上有多個GPU怜姿。
采用data=data.cuda(1)的方式可以把數(shù)據(jù)轉(zhuǎn)存到GPU1塊上慎冤。

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市沧卢,隨后出現(xiàn)的幾起案子蚁堤,更是在濱河造成了極大的恐慌,老刑警劉巖但狭,帶你破解...
    沈念sama閱讀 218,546評論 6 507
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件披诗,死亡現(xiàn)場離奇詭異,居然都是意外死亡立磁,警方通過查閱死者的電腦和手機呈队,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 93,224評論 3 395
  • 文/潘曉璐 我一進店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來唱歧,“玉大人宪摧,你說我怎么就攤上這事粒竖。” “怎么了几于?”我有些...
    開封第一講書人閱讀 164,911評論 0 354
  • 文/不壞的土叔 我叫張陵蕊苗,是天一觀的道長。 經(jīng)常有香客問我孩革,道長岁歉,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 58,737評論 1 294
  • 正文 為了忘掉前任膝蜈,我火速辦了婚禮锅移,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘饱搏。我一直安慰自己非剃,他們只是感情好,可當我...
    茶點故事閱讀 67,753評論 6 392
  • 文/花漫 我一把揭開白布推沸。 她就那樣靜靜地躺著备绽,像睡著了一般。 火紅的嫁衣襯著肌膚如雪鬓催。 梳的紋絲不亂的頭發(fā)上肺素,一...
    開封第一講書人閱讀 51,598評論 1 305
  • 那天,我揣著相機與錄音宇驾,去河邊找鬼倍靡。 笑死,一個胖子當著我的面吹牛课舍,可吹牛的內(nèi)容都是我干的塌西。 我是一名探鬼主播,決...
    沈念sama閱讀 40,338評論 3 418
  • 文/蒼蘭香墨 我猛地睜開眼筝尾,長吁一口氣:“原來是場噩夢啊……” “哼捡需!你這毒婦竟也來了?” 一聲冷哼從身側(cè)響起筹淫,我...
    開封第一講書人閱讀 39,249評論 0 276
  • 序言:老撾萬榮一對情侶失蹤站辉,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后损姜,有當?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體庵寞,經(jīng)...
    沈念sama閱讀 45,696評論 1 314
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 37,888評論 3 336
  • 正文 我和宋清朗相戀三年薛匪,在試婚紗的時候發(fā)現(xiàn)自己被綠了。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片脓鹃。...
    茶點故事閱讀 40,013評論 1 348
  • 序言:一個原本活蹦亂跳的男人離奇死亡逸尖,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出,到底是詐尸還是另有隱情娇跟,我是刑警寧澤岩齿,帶...
    沈念sama閱讀 35,731評論 5 346
  • 正文 年R本政府宣布,位于F島的核電站苞俘,受9級特大地震影響盹沈,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜吃谣,卻給世界環(huán)境...
    茶點故事閱讀 41,348評論 3 330
  • 文/蒙蒙 一乞封、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧岗憋,春花似錦肃晚、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 31,929評論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至监徘,卻和暖如春晋修,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背凰盔。 一陣腳步聲響...
    開封第一講書人閱讀 33,048評論 1 270
  • 我被黑心中介騙來泰國打工墓卦, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人廊蜒。 一個月前我還...
    沈念sama閱讀 48,203評論 3 370
  • 正文 我出身青樓趴拧,卻偏偏與公主長得像,于是被迫代替她去往敵國和親山叮。 傳聞我的和親對象是個殘疾皇子著榴,可洞房花燭夜當晚...
    茶點故事閱讀 44,960評論 2 355