Pytorch 如何查看、釋放已關(guān)閉程序占用的GPU資源
看代碼吧~
import torch print(torch.cuda.current_device()) print(torch.cuda.device_count()) print(torch.cuda.get_device_name()) print(torch.cuda.is_available())
打開(kāi)terminal輸入nvidia-smi可以看到當(dāng)前各個(gè)顯卡及用戶使用狀況，如下圖所示，使用kill -9 pid(需替換成具體的編號(hào))即可殺掉占用資源的程序，殺完后結(jié)果如下圖所示，可以發(fā)現(xiàn)再也沒(méi)有對(duì)應(yīng)自己的程序了！
補(bǔ)充一下師弟幫忙的記錄截圖，方便以后查詢使用：
補(bǔ)充：如何處理Pytorch使用GPU后仍有GPU資源未釋放的情況
使用PyTorch設(shè)置多線程（threads）進(jìn)行數(shù)據(jù)讀?。―ataLoader），其實(shí)是假的多線程，他是開(kāi)了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)文章
在本篇文章里，小編給大家分享的是關(guān)于Python安裝第三方模塊的方法及實(shí)例代碼，需要的朋友們可以學(xué)習(xí)下。2020-05-05
今天小編就為大家分享一篇python 使用遞歸回溯完美解決八皇后的問(wèn)題，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2020-02-02
Python計(jì)算一個(gè)點(diǎn)到所有點(diǎn)的歐式距離實(shí)現(xiàn)方法
今天小編就為大家分享一篇Python計(jì)算一個(gè)點(diǎn)到所有點(diǎn)的歐式距離實(shí)現(xiàn)方法，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2019-07-07
10分鐘教你用python動(dòng)畫(huà)演示深度優(yōu)先算法搜尋逃出迷宮的路徑
這篇文章主要介紹了10分鐘教你用python動(dòng)畫(huà)演示深度優(yōu)先算法搜尋逃出迷宮的路徑,非常不錯(cuò)，具有一定的參考借鑒價(jià)值，需要的朋友可以參考下2019-08-08
Django用戶認(rèn)證系統(tǒng)如何實(shí)現(xiàn)自定義
這篇文章主要介紹了Django用戶認(rèn)證系統(tǒng)如何實(shí)現(xiàn)自定義,文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2020-11-11
python 遍歷pd.Series的index和value
今天小編就為大家分享一篇python 遍歷pd.Series的index和value，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2019-11-11
解決Keras 中加入lambda層無(wú)法正常載入模型問(wèn)題
這篇文章主要介紹了解決Keras 中加入lambda層無(wú)法正常載入模型問(wèn)題，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2020-06-06
這篇文章主要介紹了python爬蟲(chóng)請(qǐng)求頭的使用,文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2020-12-12
這篇文章主要為大家詳細(xì)介紹了Python實(shí)現(xiàn)LRU緩存置換算法，文中示例代碼介紹的非常詳細(xì)，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2022-05-05
今天小編就為大家分享一篇python抓取網(wǎng)站的圖片并下載到本地的方法，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2018-05-05