当前位置:首页 > 服务端 > 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})

作者:半个王子
来源链接:https://www.cnblogs.com/zengcv/p/6956359.html

版权声明:
1、Java侠(https://www.javaxia.com)以学习交流为目的,由作者投稿、网友推荐和小编整理收藏优秀的IT技术及相关内容,包括但不限于文字、图片、音频、视频、软件、程序等,其均来自互联网,本站不享有版权,版权归原作者所有。

2、本站提供的内容仅用于个人学习、研究或欣赏,以及其他非商业性或非盈利性用途,但同时应遵守著作权法及其他相关法律的规定,不得侵犯相关权利人及本网站的合法权利。
3、本网站内容原作者如不愿意在本网站刊登内容,请及时通知本站(javaclubcn@163.com),我们将第一时间核实后及时予以删除。





本文链接:https://www.javaxia.com/server/125562.html

标签: out of memory
分享给朋友:

“Tensorflow常见报错处理” 的相关文章

[C#][控件]常用控件命名规范2022年05月17日 20:45:24
计算机组成原理考试复习2022年05月18日 15:56:51
MySQL安装之yum安装2022年05月19日 19:54:29
Logger打印等级设置2022年05月19日 20:04:13
Python 树表查找2022年05月20日 21:28:27
Python 查找算法2022年05月20日 21:28:36
关于设计模式2022年05月23日 23:02:13