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模拟浏览器发送http请求,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-11-11
这篇文章主要介绍了Python实现PS滤镜的旋转模糊功能,涉及Python基于skimage库针对图片进行旋转与模糊化处理的相关操作技巧,需要的朋友可以参考下2018-01-01
手把手教你如何用Pycharm2020.1.1配置远程连接的详细步骤
这篇文章主要介绍了如何用Pycharm2020.1.1配置远程连接，分步骤给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友参考下吧2020-08-08
在Python中,dataclass是一种用于快速创建数据类的装饰器和工具,本文实例代码中我们定义了一个Person数据类,并使用fields()函数遍历其字段,打印出每个字段的名称、类型、默认值和元数据,对python dataclass 数据类相关知识感兴趣的朋友一起看看吧2024-03-03
这篇文章主要介绍了python 实现图片批量压缩的示例，帮助大家更好的利用python处理图片，感兴趣的朋友可以了解下2020-12-12
Python3.5基础之函数的定义与使用实例详解【参数、作用域、递归、重载等】
这篇文章主要介绍了Python3.5基础之函数的定义与使用,结合实例形式详细分析了Python3.5函数的定义、参数、作用域、递归、重载、内置函数等基本概念与相关使用技巧,需要的朋友可以参考下2019-04-04
这篇文章主要介绍了python实现一般游戏的自动点击,本文给大家分享具体操作代码及需要的软件，需要的朋友可以参考下2021-10-10
今天带大家复习Python基础知识,文中对Python常用的内置对象作了非常详细的介绍及代码示例,对正在学习python的小伙伴们很有帮助,需要的朋友可以参考下2021-05-05
在本篇文章中，我们将深入探讨Pandas库中两个重要的数据处理功能：处理缺失数据和数据聚合，文中有详细的代码示例，对我们的学习或工作有一定的帮助,需要的朋友可以参考下2023-07-07
本文给大家介绍的是Python数据类型中的元祖（tuple）,简单的说Tuple,与列表一样，元素也是不可变的，但与列表不同，在一个元祖可以包含不同类型的元素2016-05-05
最新评论