CUDA_ERROR_OUT_OF_MEMORY張量流

Question

作為研究項目的一部分，我嘗試訓練一個神經網絡，該網絡對圖像進行分割（基於FCN），在執行過程中，我收到以下錯誤消息：

tensorflow.python.framework.errors_impl.ResourceExhaustedError：在分配帶有shape [1,67,1066,718]的張量並在/ job：localhost / replica：0 / task：0 / device：GPU：0上通過分配器GPU_0_bfc在type上鍵入float

請注意，我已將batch_size固定為1，即使嘗試不同的圖像尺寸，我也有相同的錯誤，我也只訓練了1張圖像，而不是1600錯誤。 你能幫我解決這個問題嗎？ 到底是什么意思？

Answer 1

歡迎使用Stackoverflow。

您應該檢查GPU和可用內存。

這不是張量流的問題。 但是您的圖形卡太小。

您可能必須使用內存需求較低的網絡或較大的圖形卡。 這不僅與您放入的圖像大小有關，而且所有權重也都需要存儲在GPU上。

如果您需要使用此特定網絡，並且不想購買另一台PC，則可以使用CloudService，例如AWS或GoogleCloudPlatform。

要檢查您的GPU狀態，您可以使用nvidia-smi