解決Python中指定GPU不生效的問(wèn)題

在做實(shí)驗(yàn)的過(guò)程中缩滨,想指定GPU卡來(lái)跑葵孤,用了好多方法都不生效咏尝。比如:
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "2,3"
按照上面方法指定后茎匠,依然是在GPU:0上跑岭辣,原因在于鼻听,應(yīng)該在最開始加載包的地方指定财著。
比如
import os
import torch
os.environ["CUDA_VISIBLE_DEVICES"] = "2,3"
這種設(shè)置是無(wú)效的,需要更改為:
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "2,3"
import torch
此時(shí)就解決了撑碴。

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
  • 序言:七十年代末撑教,一起剝皮案震驚了整個(gè)濱河市,隨后出現(xiàn)的幾起案子醉拓,更是在濱河造成了極大的恐慌伟姐,老刑警劉巖收苏,帶你破解...
    沈念sama閱讀 219,039評(píng)論 6 508
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場(chǎng)離奇詭異愤兵,居然都是意外死亡鹿霸,警方通過(guò)查閱死者的電腦和手機(jī),發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 93,426評(píng)論 3 395
  • 文/潘曉璐 我一進(jìn)店門秆乳,熙熙樓的掌柜王于貴愁眉苦臉地迎上來(lái)懦鼠,“玉大人,你說(shuō)我怎么就攤上這事屹堰「鹈疲” “怎么了?”我有些...
    開封第一講書人閱讀 165,417評(píng)論 0 356
  • 文/不壞的土叔 我叫張陵双藕,是天一觀的道長(zhǎng)淑趾。 經(jīng)常有香客問(wèn)我,道長(zhǎng)忧陪,這世上最難降的妖魔是什么扣泊? 我笑而不...
    開封第一講書人閱讀 58,868評(píng)論 1 295
  • 正文 為了忘掉前任,我火速辦了婚禮嘶摊,結(jié)果婚禮上延蟹,老公的妹妹穿的比我還像新娘。我一直安慰自己叶堆,他們只是感情好阱飘,可當(dāng)我...
    茶點(diǎn)故事閱讀 67,892評(píng)論 6 392
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著虱颗,像睡著了一般沥匈。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上忘渔,一...
    開封第一講書人閱讀 51,692評(píng)論 1 305
  • 那天高帖,我揣著相機(jī)與錄音,去河邊找鬼畦粮。 笑死散址,一個(gè)胖子當(dāng)著我的面吹牛,可吹牛的內(nèi)容都是我干的宣赔。 我是一名探鬼主播预麸,決...
    沈念sama閱讀 40,416評(píng)論 3 419
  • 文/蒼蘭香墨 我猛地睜開眼,長(zhǎng)吁一口氣:“原來(lái)是場(chǎng)噩夢(mèng)啊……” “哼儒将!你這毒婦竟也來(lái)了吏祸?” 一聲冷哼從身側(cè)響起,我...
    開封第一講書人閱讀 39,326評(píng)論 0 276
  • 序言:老撾萬(wàn)榮一對(duì)情侶失蹤椅棺,失蹤者是張志新(化名)和其女友劉穎犁罩,沒(méi)想到半個(gè)月后齐蔽,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 45,782評(píng)論 1 316
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡床估,尸身上長(zhǎng)有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 37,957評(píng)論 3 337
  • 正文 我和宋清朗相戀三年含滴,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片丐巫。...
    茶點(diǎn)故事閱讀 40,102評(píng)論 1 350
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡谈况,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出递胧,到底是詐尸還是另有隱情碑韵,我是刑警寧澤,帶...
    沈念sama閱讀 35,790評(píng)論 5 346
  • 正文 年R本政府宣布缎脾,位于F島的核電站祝闻,受9級(jí)特大地震影響,放射性物質(zhì)發(fā)生泄漏遗菠。R本人自食惡果不足惜联喘,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 41,442評(píng)論 3 331
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望辙纬。 院中可真熱鬧豁遭,春花似錦、人聲如沸贺拣。這莊子的主人今日做“春日...
    開封第一講書人閱讀 31,996評(píng)論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽(yáng)譬涡。三九已至闪幽,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間昂儒,已是汗流浹背沟使。 一陣腳步聲響...
    開封第一講書人閱讀 33,113評(píng)論 1 272
  • 我被黑心中介騙來(lái)泰國(guó)打工委可, 沒(méi)想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留渊跋,地道東北人。 一個(gè)月前我還...
    沈念sama閱讀 48,332評(píng)論 3 373
  • 正文 我出身青樓着倾,卻偏偏與公主長(zhǎng)得像拾酝,于是被迫代替她去往敵國(guó)和親。 傳聞我的和親對(duì)象是個(gè)殘疾皇子卡者,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 45,044評(píng)論 2 355

推薦閱讀更多精彩內(nèi)容