Tensorflow,Keras使用动态显存分配

使用tf,keras训练时常常占满显存,因此采用动态显存分配策略节约资源

import tensorflow as tf #这里如果用到tf2版本,则替换为> import tensorflow.compat.v1 as tf
config = tf.ConfigProto()
config.gpu_options.allow_growth=True
sess = tf.Session(config=config)

Guess you like

Origin blog.csdn.net/qi_sama/article/details/121943794