显示当前正在使用的 GPU 的 ID

显示当前正在使用的 GPU 的 ID

我的服务器中有 4 张 GPU 卡,我可以通过 nvidia-smi 查看哪些卡正在运行,并检查内存是否已满,但我想自动执行此过程,是否有任何命令可以返回例如正在使用的 GPU 设备 ID = 0,1,2,而不是目测检查 nvidia-smi?谢谢!

相关内容