Debug --> 奇奇怪怪的显卡调用错误

1 Function call stack:
2 train_function -> train_function -> train_function

解决办法:

1 import tensorflow as tf
2 import keras
3 config=tf.compat.v1.ConfigProto()
4 config.gpu_options.allow_growth = True
5 sess=tf.compat.v1.Session(config=config)
6 keras.backend.tensorflow_backend.set_session(sess)
To see I can not see, to know I do not know.
原文地址:https://www.cnblogs.com/aluomengmengda/p/14686393.html