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)

 

posted @ 2021-04-21 18:33  阿洛萌萌哒  阅读(428)  评论(0)  编辑  收藏  举报