如何通过pytorch获得可用的空闲GPU?

我们有一个 GPU 服务器,其中有几张 GPU 卡,由几位工程师共享。如何找到哪个 GPU 仍处于空闲状态以便我可以将其用于我的工作?我知道 nvidia-smi 命令可以在控制台中完成,但是 pytorch 代码是否有可能找到哪个 GPU 仍然空闲?

谢谢。

rrrttt1984 回答:如何通过pytorch获得可用的空闲GPU?

暂时没有好的解决方案,如果你有好的解决方案,请发邮件至:iooj@foxmail.com
本文链接:https://www.f2er.com/221065.html

大家都在问