VGG16模型凍結計算機

Question

我目前正在嘗試使用keras庫中的vgg16模型，但是每當我通過執行以下操作創建VGG16模型的對象時，

from keras.applications.vgg16 import VGG16
model = VGG16()

我收到以下消息3次。

tensorflow/core/framework/allocator.cc.124 allocation of 449576960 exceeds 10% of system memory

之后，我的電腦死機了。 我在Linux Mint 18上使用64位4GB RAM，但無法訪問GPU。

這個問題與我的RAM有關嗎？

作為一個臨時解決方案，我從命令行運行python腳本，因為與任何IDE相比，我的計算機在那里凍結的次數更少。 另外，當我使用任何其他模型（例如InceptionV3）時，也不會發生這種情況。

我已經嘗試過這里提供的解決方案

但這沒用

任何幫助表示贊賞。

Answer 1

您很可能用完了內存（RAM）。 嘗試並行運行top（或htop）並查看您的內存利用率。

通常，VGG模型很大，並且需要相當數量的RAM。 也就是說，實際需求取決於批量大小。 較小的批次意味着較小的活化層。

例如，一個6幅圖像的批處理將消耗大約一公羊的內存（ 參考 ）。 作為測試，您可以將批量大小減小到1，然后查看適合您的內存的大小。