1. ホーム

Keras/TensorFlowのエラーです。CUDA_ERROR_OUT_OF_MEMORY 解決策

2022-02-12 01:01:45

Keras/TensorFlowは以下のエラーを報告します。

ホスト上で2097152バイトの割り当てに失敗しました: CUDA_ERROR_OUT_OF_MEMORY

サイズ:xxxxxxxのピン留めされたホストメモリを割り当てられませんでした。

回避策

TensorFlowはデフォルトで貪欲にビデオメモリを占有するため、ビデオメモリが足りない場合があります。以下のコードを追加して、オンデマンドでビデオメモリを利用できるようにします。

import tensorflow as tf
import keras.backend.tensorflow_backend as KTF

config = tf.ConfigProto()  
config.gpu_options.allow_growth=True # don't fill up memory, allocate as needed
sess = tf.Session(config=config)
KTF.set_session(sess) # set session

問題は解決し、メモリが常に満杯になるわけではありません