2015-09-28 49 views
0

我有一個Theano正在使用批量數據的代碼。當我增加批量大小時,它會拋出SegFault錯誤。它工作得很好,達到batch size = 750。但increasing1000是扔我錯誤。我還檢查了它在任何時候只使用50MB of GPU memory。但我的系統上有128MB GPU memory。任何人都可以幫我在這裏調試問題。Theano正在拋出分段錯誤

+0

您是如何測量GPU內存使用量的?請注意,GPU驅動程序和Theano都有一些使用某些GPU內存的開銷。 128MB是非常少量的GPU內存,這些開銷很容易在78MB的範圍內。 –

+0

使用nvidia-smi – neel

+0

內存使用在批量大小中是線性的嗎?例如如果您測量批量大小爲「100,200,300,400,500,600,700」的內存並繪製結果,那麼這些點是否按直線排列?如果不是,並且你推斷'batch size = 1000',它是否超過128MB? –

回答

0

您可以通過啓用profileprofile_memory配置標誌來剖析Theano的內存使用情況。有關更多詳細信息,請參閱documentation

還有關於Python/Theano內存管理的更多信息here