最新国产好看的视频,伊人天堂AV在线,国产Aaaaaa视频,蜜臀视频在线观看一区,人妻av色图,密臀久久久精品影片,青青视频免费观看毛片,久草在线观看视,国产三级精品色情在线

Pytorch 如何查看、釋放已關(guān)閉程序占用的GPU資源

 更新時(shí)間:2021年05月20日 14:03:31   作者:JianzhuWang  
這篇文章主要介紹了Pytorch 查看、釋放已關(guān)閉程序占用的GPU資源的操作,具有很好的參考價(jià)值,希望對大家有所幫助。

看代碼吧~

import torch 
print(torch.cuda.current_device())
print(torch.cuda.device_count())
print(torch.cuda.get_device_name())
print(torch.cuda.is_available())

打開terminal輸入nvidia-smi可以看到當(dāng)前各個(gè)顯卡及用戶使用狀況,如下圖所示,使用kill -9 pid(需替換成具體的編號(hào))即可殺掉占用資源的程序,殺完后結(jié)果如下圖所示,可以發(fā)現(xiàn)再也沒有對應(yīng)自己的程序了!

補(bǔ)充一下師弟幫忙的記錄截圖,方便以后查詢使用:

補(bǔ)充:如何處理Pytorch使用GPU后仍有GPU資源未釋放的情況

使用PyTorch設(shè)置多線程(threads)進(jìn)行數(shù)據(jù)讀?。―ataLoader),其實(shí)是假的多線程,他是開了N個(gè)子進(jìn)程(PID都連著)進(jìn)行模擬多線程工作,所以你的程序跑完或者中途kill掉主進(jìn)程的話,子進(jìn)程的GPU顯存并不會(huì)被釋放,需要手動(dòng)一個(gè)一個(gè)kill才行

具體方法描述如下:

1.先關(guān)閉ssh(或者shell)窗口,退出重新登錄

2.查看運(yùn)行在gpu上的所有程序:

fuser -v /dev/nvidia*

3.kill掉所有(連號(hào)的)僵尸進(jìn)程

以上為個(gè)人經(jīng)驗(yàn),希望能給大家一個(gè)參考,也希望大家多多支持腳本之家。

相關(guān)文章

最新評論

衡阳市| 麻江县| 邢台市| 平南县| 邹平县| 阳信县| 集贤县| 水富县| 法库县| 翼城县| 宁陵县| 邓州市| 武汉市| 章丘市| 长宁县| 澜沧| 大邑县| 永和县| 长寿区| 文安县| 册亨县| 土默特左旗| 溧阳市| 鸡东县| 扎赉特旗| 邹城市| 鲁山县| 土默特左旗| 公主岭市| 朝阳市| 土默特左旗| 民乐县| 益阳市| 梁山县| 舒城县| 伊春市| 西安市| 邻水| 长顺县| 开远市| 南城县|