tensorflow调用GPU时报错:
再模型调用前加上如下代码:
config = tf.ConfigProto(gpu_options=tf.GPUOptions(allow_growth=True))
sess = tf.Session(config=config)
自己的代码是:
config = tf.ConfigProto(allow_soft_placement=True)
sess = tf.Session(config=config)
tensorflow调用GPU时报错:
再模型调用前加上如下代码:
config = tf.ConfigProto(gpu_options=tf.GPUOptions(allow_growth=True))
sess = tf.Session(config=config)
自己的代码是:
config = tf.ConfigProto(allow_soft_placement=True)
sess = tf.Session(config=config)