caffe測試時報錯 [....syncedmem.cpp:56] Check failed: error == cudaSuccess (2 vs. 0) out of memory

現象: 在caffe復現DenseNet的過程中,訓練完成了,就到了測試階段。當開始運行之後,過了一段時間發現報了一個錯誤。 F0523 12:55:04.591845 17608 syncedmem.cpp:56] Check failed: error == cudaSuccess (2 vs. 0) out of memory。爆內存了。如圖所示。 分析一:batch_size 一般這個問題
相關文章
相關標籤/搜索