在用pytorch训练完模型后,需要将显存手动释放掉
使用nvidia-smi
查看占用显存的进程
kill -9 PID(进程号)
来结束进程,即可释放显存
释放显存
最新推荐文章于 2023-07-19 11:45:34 发布
在用pytorch训练完模型后,需要将显存手动释放掉
使用nvidia-smi
查看占用显存的进程
kill -9 PID(进程号)
来结束进程,即可释放显存