我的代码中,产生这个错误的解决办法是添加
config = tf.ConfigProto(gpu_options=tf.GPUOptions(allow_growth=True))
sess = tf.Session(config=config)
限制显存的增长,不一定适用全部~
我的代码中,产生这个错误的解决办法是添加
config = tf.ConfigProto(gpu_options=tf.GPUOptions(allow_growth=True))
sess = tf.Session(config=config)
限制显存的增长,不一定适用全部~