【Matlab】Matlab2016a-'Out of memory on device' error for GPU

平臺: Matlab 2016a+Win10+GPU 出錯情況:採用Matconvnet在matlab中訓練網絡,train的過程中報出’Out of memory on device’ 的錯誤,具體如下: 解決方法:經檢查發現train的代碼中batchsize設置爲64過高,改爲32後正常運行。
相關文章
相關標籤/搜索