Ctrl+C 中止运行后 GPU 存储没有及时释放,需要手动清空
方法一:
在 PyTorch 内部可以
torch.cuda.empty_cache()
方法二:
在命令行可以先使用 ps 找到程序的 PID,再使用 kill 结束该进程
ps aux | grep python
kill -9 [pid]
方法三:
直接重置没有被清空的 GPU
nvidia-smi --gpu-reset -i [gpu_id]
Ctrl+C 中止运行后 GPU 存储没有及时释放,需要手动清空
方法一:
在 PyTorch 内部可以
torch.cuda.empty_cache()
方法二:
在命令行可以先使用 ps 找到程序的 PID,再使用 kill 结束该进程
ps aux | grep python
kill -9 [pid]
方法三:
直接重置没有被清空的 GPU
nvidia-smi --gpu-reset -i [gpu_id]
本文标题:清除GPU占用
本文链接:https://www.haomeiwen.com/subject/ckusfltx.html
网友评论