首页 > 代码库 > Tensorflow常见报错处理
Tensorflow常见报错处理
1. CUDA_ERROR_OUT_OF_MEMORY
import tensorflow as tf
# 避免 CUDA_ERROR_OUT_OF_MEMORY
config = tf.ConfigProto(allow_soft_placement=True)
gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.7)
config.gpu_options.allow_growth = True
sess = tf.Session(config=config)
a = tf.placeholder("float")
b = tf.placeholder("float")
c = tf.constant(6.0)
d = tf.multiply(a, b)
y = tf.multiply(d, c)
print sess.run(y, feed_dict={a: 3, b: 3})
Tensorflow常见报错处理
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。