Pytorch GPU運算過程中會出現:“cuda runtime error(2): out of memory”這樣的錯誤。通常,這種錯誤是由于在循環中使用全局變量當做累加器,且累加梯度信息的緣故
億速云公眾號
手機網站二維碼
Copyright ? Yisu Cloud Ltd. All Rights Reserved. 2018 版權所有
廣州億速云計算有限公司粵ICP備17096448號 粵公網安備 44010402001142號增值電信業務經營許可證編號:B1-20181529