贊助廠商

娛樂城推薦

首頁

刊登資訊

  • 刊登者:匿名
  • 時間:2021-06-09 08:50:09

尚未解答DataScience- Pytorch記憶體問題

DataScience- Pytorch記憶體問題

想要請問版上各位pytorch的使用者會不會有記憶體管理的問題呢

我目前遇到的狀況是存下來的pth file約800MB 包含了optimizer之類的

所有關於training的參數與超參數,但是load進去環境裡面之後立刻就佔了2.4GB

我相信是會有一些overhead沒錯,但是三倍這個數字好像超乎我的理解之外了

接著是一張4000 * 6000的圖片就會佔100MB左右的記憶體,當我call

model(img)之後,記憶體的使用量會瞬間暴增5GB,我有點不太懂這裡面到底

發生甚麼事,又應該要做甚麼來解決呢

我目前已經是有call .eval()也有在with torch.no_grad()裡面作業了

但是還是只能一次inference一張圖片,結果gpu的效果大打折扣

有沒有人遇到類似的情況或者知道該怎麼解決這個問題呢

--

0個答案 DataScience- Pytorch記憶體問題

其他問題

友站連結