在jupyter notebook上训练模型gpu显存不够，关闭后模型仍占有显存的坑

分类: 文章 • 2022-10-27 23:39:56

在任务管理器的性能一栏能看到gpu的运行情况

在jupyter notebook上训练模型gpu显存不够，关闭后发现GPU仍然占有很大显存，导致后面运行不了，无论是修改batch_size还是之类的，都无效。
解决的办法，重启服务，即红色所指即可，再重新运行程序就ok了。