在jupyter notebook上訓練模型gpu顯存不夠,關閉後模型仍佔有顯存的坑

在任務管理器的性能一欄能看到gpu的運行情況 在jupyter notebook上訓練模型gpu顯存不夠,關閉後發現GPU仍然佔有很大顯存,導致後面運行不了,無論是修改batch_size還是之類的,都無效。 解決的辦法,重啓服務,即紅色所指即可,再重新運行程序就ok了。
相關文章
相關標籤/搜索