Keras + Python3.7

GANのために開発環境つくったので、メモ。

開発機

• OS : windows10
• GPU : GTX1660Ti

いれたもの

• python3.7.6
• tensorflow-2.2.0
• keras-2.3.1
• cuda10.1
• cuDNNv7.6.5

CUDNN_STATUS_INTERNAL_ERROR が出る問題

内容

https://github.com/tensorflow/tensorflow/issues/24496

対応

以下をコードに記載する。

import tensorflow as tf

gpu_devices = tf.config.experimental.list_physical_devices('GPU')
for device in gpu_devices: tf.config.experimental.set_memory_growth(device, True)