博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
解决:failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILED
阅读量:2080 次
发布时间:2019-04-29

本文共 426 字,大约阅读时间需要 1 分钟。

今天测试tensorflow的GPU版本,但是发现一个问题同一个程序(卷积神经网络)为什么在anaconda里的jupyter里就可以飞速跑完,在使用anaconda环境的pycharm里面就会说显存不足呢?

报错是: failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILED(这就是显存不足的意思)

 

两个解决方案:

(1)

 jupyter在没关闭的时候,会占用电脑大量显存,不应该和pycharm同时开启,应该及时按Ctrl+cl两次关闭。

(2)

在显存不足(如开了jupyter)的时候定义占比,强行跑程序。
gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.333)with tf.Session(config=tf.ConfigProto(gpu_options=gpu_options)) as sess:

转载地址:http://ohkqf.baihongyu.com/

你可能感兴趣的文章
MySQL学习笔记——20170809
查看>>
MySQL学习笔记——20170810
查看>>
MySQL学习笔记——20170811
查看>>
MySQL学习笔记——20170812
查看>>
Android内存溢出与优化(五)——防止static引用
查看>>
Scala学习笔记——20170817
查看>>
Scala学习笔记——20170818
查看>>
Scala学习笔记——20170819
查看>>
Scala学习笔记——20170820
查看>>
Python学习笔记——20170821
查看>>
Python学习笔记——20170822
查看>>
Python学习笔记——20170823
查看>>
Python学习笔记——20170824
查看>>
Python学习笔记——20170825
查看>>
Python学习笔记——20170828
查看>>
Python学习笔记——20170830
查看>>
Python学习笔记——20170829
查看>>
Python学习笔记——20170831
查看>>
天堂图片网爬虫
查看>>
设计一个简易的订阅/分发消息的工具
查看>>