释放GPU显存

使用下面的Python代码可以快速释放GPU显存

import os
pid = list(set(os.popen('fuser -v /dev/nvidia*').read().split()))
kill_cmd = 'kill -9 ' + ' '.join(pid)
print(kill_cmd)
os.popen(kill_cmd)

版权声明:本文为ECHOSON原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。