Pytorch: 指定運行 GPU

在終端使用如下命令查看,計算機本身的 GPU 數(shù)目及使用情況。

nvidia-smi
nvidia-smi_info.png

打印出的信息代表的意思:
GPU: GPU序列號,有幾個GPU就顯示幾條。
Fan:風(fēng)扇轉(zhuǎn)速,從0到100%之間變動,這個速度是計算機期望的風(fēng)扇轉(zhuǎn)速,實際情況下如果風(fēng)扇堵轉(zhuǎn),可能打不到顯示的轉(zhuǎn)速
Temp: 是溫度,單位攝氏度。
Perf: 性能狀態(tài),從P0到P12,P0表示最大性能,P12表示狀態(tài)最小性能。
Pwr:Usage/Cap: 是能耗。
Persistence-M: 是持續(xù)模式的狀態(tài),持續(xù)模式雖然耗能大,但是在新的GPU應(yīng)用啟動時,花費的時間更少。
Bus-Id: 涉及GPU總線的東西
Disp.A: Display Active,表示GPU的顯示是否初始化。
Memory-Usage: 顯存使用率。
GPU-Util: GPU利用率

然后根據(jù)GPU數(shù)目可以指定程序運行使用哪塊 GPU,默認程序使用的是 GPU 0, 我們可以在代碼最前面添加如下指令指定使用的GPU。

import os 
os.environ["CUDA_VISIBLE_DEVICES"] = "1"
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容