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())
打開terminal輸入nvidia-smi可以看到當(dāng)前各個(gè)顯卡及用戶使用狀況，如下圖所示，使用kill -9 pid(需替換成具體的編號(hào))即可殺掉占用資源的程序，殺完后結(jié)果如下圖所示，可以發(fā)現(xiàn)再也沒有對(duì)應(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)文章
基于Python模擬瀏覽器發(fā)送http請(qǐng)求
這篇文章主要介紹了基于Python模擬瀏覽器發(fā)送http請(qǐng)求,文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2020-11-11
Python實(shí)現(xiàn)PS濾鏡的旋轉(zhuǎn)模糊功能示例
這篇文章主要介紹了Python實(shí)現(xiàn)PS濾鏡的旋轉(zhuǎn)模糊功能,涉及Python基于skimage庫(kù)針對(duì)圖片進(jìn)行旋轉(zhuǎn)與模糊化處理的相關(guān)操作技巧,需要的朋友可以參考下2018-01-01
手把手教你如何用Pycharm2020.1.1配置遠(yuǎn)程連接的詳細(xì)步驟
這篇文章主要介紹了如何用Pycharm2020.1.1配置遠(yuǎn)程連接，分步驟給大家介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友參考下吧2020-08-08
python dataclass 快速創(chuàng)建數(shù)據(jù)類的方法
在Python中,dataclass是一種用于快速創(chuàng)建數(shù)據(jù)類的裝飾器和工具,本文實(shí)例代碼中我們定義了一個(gè)Person數(shù)據(jù)類,并使用fields()函數(shù)遍歷其字段,打印出每個(gè)字段的名稱、類型、默認(rèn)值和元數(shù)據(jù),對(duì)python dataclass 數(shù)據(jù)類相關(guān)知識(shí)感興趣的朋友一起看看吧2024-03-03
這篇文章主要介紹了python 實(shí)現(xiàn)圖片批量壓縮的示例，幫助大家更好的利用python處理圖片，感興趣的朋友可以了解下2020-12-12
Python3.5基礎(chǔ)之函數(shù)的定義與使用實(shí)例詳解【參數(shù)、作用域、遞歸、重載等】
這篇文章主要介紹了Python3.5基礎(chǔ)之函數(shù)的定義與使用,結(jié)合實(shí)例形式詳細(xì)分析了Python3.5函數(shù)的定義、參數(shù)、作用域、遞歸、重載、內(nèi)置函數(shù)等基本概念與相關(guān)使用技巧,需要的朋友可以參考下2019-04-04
python實(shí)現(xiàn)一般游戲的自動(dòng)點(diǎn)擊具體操作
這篇文章主要介紹了python實(shí)現(xiàn)一般游戲的自動(dòng)點(diǎn)擊,本文給大家分享具體操作代碼及需要的軟件，需要的朋友可以參考下2021-10-10
今天帶大家復(fù)習(xí)Python基礎(chǔ)知識(shí),文中對(duì)Python常用的內(nèi)置對(duì)象作了非常詳細(xì)的介紹及代碼示例,對(duì)正在學(xué)習(xí)python的小伙伴們很有幫助,需要的朋友可以參考下2021-05-05
深入探究python中Pandas庫(kù)處理缺失數(shù)據(jù)和數(shù)據(jù)聚合
在本篇文章中，我們將深入探討Pandas庫(kù)中兩個(gè)重要的數(shù)據(jù)處理功能：處理缺失數(shù)據(jù)和數(shù)據(jù)聚合，文中有詳細(xì)的代碼示例，對(duì)我們的學(xué)習(xí)或工作有一定的幫助,需要的朋友可以參考下2023-07-07
Python數(shù)據(jù)類型詳解（三）元祖：tuple
本文給大家介紹的是Python數(shù)據(jù)類型中的元祖（tuple）,簡(jiǎn)單的說Tuple,與列表一樣，元素也是不可變的，但與列表不同，在一個(gè)元祖可以包含不同類型的元素2016-05-05