Pytorch 如何查看、释放已关闭程序占用的GPU资源
更新时间：2021年05月20日 14:03:31 作者：JianzhuWang
这篇文章主要介绍了Pytorch 查看、释放已关闭程序占用的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可以看到当前各个显卡及用户使用状况，如下图所示，使用kill -9 pid(需替换成具体的编号)即可杀掉占用资源的程序，杀完后结果如下图所示，可以发现再也没有对应自己的程序了！
补充一下师弟帮忙的记录截图，方便以后查询使用：
补充：如何处理Pytorch使用GPU后仍有GPU资源未释放的情况
使用PyTorch设置多线程（threads）进行数据读取（DataLoader），其实是假的多线程，他是开了N个子进程（PID都连着）进行模拟多线程工作，所以你的程序跑完或者中途kill掉主进程的话，子进程的GPU显存并不会被释放，需要手动一个一个kill才行
具体方法描述如下：
1.先关闭ssh（或者shell）窗口，退出重新登录
2.查看运行在gpu上的所有程序：
fuser -v /dev/nvidia*
3.kill掉所有（连号的）僵尸进程
以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
在本篇文章里，小编给大家分享的是关于Python安装第三方模块的方法及实例代码，需要的朋友们可以学习下。2020-05-05
今天小编就为大家分享一篇python 使用递归回溯完美解决八皇后的问题，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-02-02
今天小编就为大家分享一篇Python计算一个点到所有点的欧式距离实现方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-07-07
10分钟教你用python动画演示深度优先算法搜寻逃出迷宫的路径
这篇文章主要介绍了10分钟教你用python动画演示深度优先算法搜寻逃出迷宫的路径,非常不错，具有一定的参考借鉴价值，需要的朋友可以参考下2019-08-08
这篇文章主要介绍了Django用户认证系统如何实现自定义,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-11-11
python 遍历pd.Series的index和value
今天小编就为大家分享一篇python 遍历pd.Series的index和value，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-11-11
这篇文章主要介绍了解决Keras 中加入lambda层无法正常载入模型问题，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-06-06
这篇文章主要介绍了python爬虫请求头的使用,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2020-12-12
这篇文章主要为大家详细介绍了Python实现LRU缓存置换算法，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2022-05-05
今天小编就为大家分享一篇python抓取网站的图片并下载到本地的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-05-05
最新评论