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

在任务管理器的性能一栏能看到gpu的运行情况 在jupyter notebook上训练模型gpu显存不够,关闭后发现GPU仍然占有很大显存,导致后面运行不了,无论是修改batch_size还是之类的,都无效。 解决的办法,重启服务,即红色所指即可,再重新运行程序就ok了。
相关文章
相关标签/搜索