[TensorFlow] 解決CUDA內部錯誤

前些時間跑TensorFlow代碼的時候出現CUDNN_STATUS_INTERNAL_ERROR錯誤
據說會有此錯誤是因為顯存分配問題所導致的BUG，加上以下代碼即可：

config = tf.ConfigProto()
config.gpu_options.allow_growth = True
sess = tf.Session(graph=graph, config=config)