Tensorflow2.0 GPUconfig的問題

今天在把之前NMT代碼遷到tf2.0 stable 上遇到了一個"CUDNN_STATUS_INTERNAL_ERROR"的問題，簡單記錄一下。
是一個跟gpu顯存有關的問題，指定gpu動態分配便可。

Tensorflow2.0中gpuconfig相關的搬到了tf.config.experimental下
document

physical_devices = tf.config.experimental.list_physical_devices('GPU')
assert len(physical_devices) > 0, "Not enough GPU hardware devices available"
tf.config.experimental.set_memory_growth(physical_devices[0], True)

放在網絡initialize前便可

github相關issue