清除GPU占用
2021-02-24 本文已影响0人
blair_liu
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]
