解决：failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILED

最新推荐文章于 2024-01-03 09:18:45 发布

瓜波牛排

最新推荐文章于 2024-01-03 09:18:45 发布

阅读量8.4k

点赞数 4

分类专栏：深度学习

本文链接：https://blog.csdn.net/Amigo_1997/article/details/88832587

版权

41 篇文章 26 订阅

订阅专栏

今天测试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:

关注