E tensorflow/stream

编程入门行业动态更新时间:2024-10-22 12:22:50

E <a href=https://www.elefans.com/category/jswz/34/1769423.html style= tensorflow/stream"/>

E tensorflow/stream

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)

更多推荐

E tensorflow/stream

本文发布于:2024-03-05 05:03:26，感谢您对本站的认可！

本文链接:https://www.elefans.com/category/jswz/34/1711374.html