赞
踩
在用pytorch训练完模型后,需要将显存手动释放掉 使用nvidia-smi查看占用显存的进程 kill -9 PID(进程号)来结束进程,即可释放显存
nvidia-smi
kill -9 PID(进程号)