Windows版Tensorflow显存不足解决方案 (2017-07-23 00:24:27)

在Windows版Tensorflow运行的时候提示错误:

failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILED

在官方Github上找到一个解决方案

创建session的时候允许显存增长

config = tf.ConfigProto()

config.gpu_options.allow_growth = True

with tf.Session(config = config) as sess:

©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

  • 问题1: pip安装时,提示找不到对应的版本“No matching distribution found ”c:...
    董春磊阅读 3,944评论 0 1
  • 1、tf.shape(varA)和varA.get_shape()这两个API都是返回varA的size大小,但是...
    溪奇阅读 4,020评论 0 1
  • 你提到有几次来访者“脱落”的经历,似乎对自己和来访者的关系产生了一些怀疑,能够看出你非常在乎自己是否能够帮上对方,...
    yuzideyuzi阅读 420评论 2 2
  • 你说芰荷抖攒,熠然风华;后来泪脸青房,滚珠入画。 你说子夜悠长,暗香扰窗;后来白水月华,及地成霜。 你说青伞桃花,...
    眉眼娇怯阅读 524评论 26 22
  • C语言今天答辩,准备了好几天的结业设计今天终于圆满完成了,得到了老师的肯定和认可,这些天的付出没有白费,再接再厉,...
    欢喜洋洋阅读 204评论 0 0

友情链接更多精彩内容