import tensorflow as tf
config = tf.compat.v1.ConfigProto()
# 0.5是指所占显存的比例
config.gpu_options.per_process_gpu_memory_fraction = 0.5
tf.compat.v1.keras.backend.set_session(tf.compat.v1.Session(config=config))
Keras2.1.0限制显存
最新推荐文章于 2023-06-05 16:09:40 发布
关键词由CSDN通过智能技术生成