使用tensorflow-gpu训练数据出现CUDNN_STATUS_ALLOC_FAILED

最新推荐文章于 2022-02-28 16:18:04 发布

Lotus_land_dream

最新推荐文章于 2022-02-28 16:18:04 发布

阅读量1.3k

点赞数 1

分类专栏： tensorflow 文章标签： tensorflow

本文链接：https://blog.csdn.net/Lotus_land_dream/article/details/89355322

版权

tensorflow 专栏收录该内容

1 篇文章

订阅专栏

本文探讨了在使用TensorFlow进行深度学习时遇到的显存不足问题，特别是在显卡性能有限的情况下。通过调整配置参数，如启用GPU显存按需增长，可以有效避免显存溢出错误，同时提供了降低batch_size作为另一种解决方案。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED

当你看到这个就证明你的显卡不怎么行了，显存不够
tensorflow的配置加上下面的代码

  config = tf.ConfigProto()
  config.gpu_options.allow_growth = True
  session = tf.Session(config=config)

Hint: If you want to see a list of allocated tensors when OOM happens
出现这个问题也是显卡太渣，改小batch_size

确定要放弃本次机会？

福利倒计时

: :

立减 ¥

普通VIP年卡可用

立即使用

Lotus_land_dream

关注关注

1
点赞
踩
1

收藏

觉得还不错? 一键收藏
0
评论
分享

复制链接

分享到 QQ

分享到新浪微博

扫一扫
举报

举报

专栏目录

Tensorflow2.0 之Could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED问题

01-20

Tensorflow2.0 之Could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED问题问题描述：在tensorflow2.0的学习过程中，遇到了Could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED，发现这个问题是我在调用kseras模块的卷积类前向传播时触发的，之前的全连接层都没有出现过类似的问题。分析问题应该是gpu没有正确的分配所导致的。 import tensorflow as tf from tensorflow.keras import layers x =

参与评论您还未登录，请先登录后发表或查看评论

使用Tensorflow-gpu出现错误:Could not create cudnn handle: CUDNN_STATUS_NOT_INITIALIZED

qq_43102387的博客

11-15

902

使用Tensorflow-gpu出现错误：错误提示 Could not create cudnn handle: CUDNN_STATUS_NOT_INITIALIZED ****Error： Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above.

mxnet CUDNN_STATUS_ALLOC_FAILED 错误 cudnn出错

insanegtp的博客

11-22

7483

一直用别人的填坑帖子搞深度学习，没有贡献，比较惭愧。。。。这是我自己填的一个坑。。。。。当运行mxnet时，用cudnn，报错：CUDNN_STATUS_ALLOC_FAILED 具体错误没记，大概是..(.. vs ..)。上网一查，发现一种说法是GPU显卡的计算因数（computablilty??忘了具体怎么拼写的了）要3以上，但我的显卡是gtx 1070，GPU计算因

运行tensorflow -gpu出现 CUDNN_STATUS_ALLOC_FAILED

耶律大石的博客

09-08

373

运行tensorflow -gpu出现 CUDNN_STATUS_ALLOC_FAILED背景介绍解决办法背景介绍我是cuda 10.2版本，用的conda install 安装的tf，使用yolo v4的时候出现这个问题。解决办法在代码最开头加入如下代码。 import tensorflow as tf config = tf.ConfigProto(gpu_options=tf.GPUOptions(allow_growth=True)) sess = tf.Session(config=co

cuDNN error: CUDNN_STATUS_ALLOC_FAILED

qq_44796370的博客

02-28

1875

显存不够，调小batch-size，立竿见影！

报错解决：could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED

あずにゃん梓喵的博客

08-25

1535

个人主页执行时报错：could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED 解决方式：GPU占用问题导致的，keras框架（Tensorflow backend）设置GPU按需分配： import tensorflow as tf from keras import backend as K config = tf.C...

【keras】CUDNN_STATUS_ALLOC_FAILED 和 failed to run cuBLAS routine cublas ，keras和tensflow-gpu版本的问题

qq_37791134的博客

03-01

2253

【报错】首先报错： could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED尝试解决： https://blog.csdn.net/weixin_44754046/article/details/97663626 在最开头添加： import tensorflow as tf config = tf.ConfigProto(gpu_o...

ExternalError: Cudnn error, CUDNN_STATUS_NOT_INITIALIZED

qq_41675931的博客

02-05

4791

项目场景： windows安装paddlepaddle-gpu出现错误环境描述：系统：win10 cuda版本：10.0.130 cudnn版本：v7.6 Nvidia 显卡驱动411.63（这个很关键） paddlepaddle版本：v1.8（gpu版，我在v1.7，v2.0也出现了这个问题） python版本：3.6 问题描述：按照飞桨（paddle）官网的安装教程一步一步来，最后验证安装的时候出现如下错误： ExternalError: Cudnn error, CUDNN_STATUS_N

问题解决：Could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED

weixin_46276803的博客

02-25

898

tensorflow2.0版本错误和cudnn有关，排除了cuda和cudnn的版本问题，在网上查了一下，应该是GPU被占用，所以无法使用，需要在代码里面添加几行语句，设置GPU的按需分配添加代码 import tensorflow as tf physical_devices = tf.config.experimental.list_physical_devices('GPU') asse...

could not create cudnn handle CUDNN_STATUS_ALLOC_FAILED

qq_36278207的博客

01-07

1401

Could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED 1.问题描述及解决在运行tensorflow-gpu的过程中出现了这一问题。网络查询解决办法，说法不一，即引起这种错误有多种可能性。这种错误为cudnn资源分配出现，以下列举常见几种报错可能性和解决办法：（a）多种软件环境同时import tensorflow。例如：Pycharm...

Tensroflow training error: CUDNN_STATUS_ALLOC_FAILED

Color Space的博客

07-08

865

详细错误信息： E T:\src\github\tensorflow\tensorflow\stream_executor\cuda\cuda_dnn.cc:352] Could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED 网上大部分是说显存问题，当然我的GPU是6G 1660Ti确实一般，给出的解决方法就是分配GPU 一般情...

Could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED的一种解决方案

He110__W0r1d的博客

11-07

747

最近在使用tensorflow-gpu2.0的时候老是报这个错误： Could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED BaseCollectiveExecutor::StartAbort Unknown: Failed to get convolution algorithm. This is probably because c...

TensorFlow 1.13.1 CUDNN_STATUS_ALLOC_FAILED 错误解决办法

chyelang12的博客

04-18

1660

为使用faceswap玩一下换脸，在win10+rtx2070平台上搭建TensorFlow-gpu 1.13.1环境。安装完毕后import tensorflow没有问题，但在运行时总是报Could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED错误，上网多方查证后认为cuda、cudnn、TensorFlow的版本搭配没有问题，但就是反复重...

【tensorflow 报错解决】not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED

RS_Deer的博客

02-26

1085

今天运行一个代码，前两天跑的好好的，突然就报下面的错误 E T:\src\github\tensorflow\tensorflow\stream_executor\cuda\cuda_dnn.cc:332] could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED E T:\src\github\tensorflow\tensorflow\stream_executor\cuda\cuda_dnn.cc:332] could not create

tensorflow-gpu error：CUDNN_STATUS_ALLOC_FAILED或者self._traceback = tf_stack.extract_stack()

Draw_1999的博客

07-08

5552

在有些情况下，因为深度学习框架版本更新，细节的变动会使我们的代码最初对应修改：报错信息（出现其中一种）： 1.Could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED 2.self._traceback = tf_stack.extract_stack() 若文本导入tensorflow:则在红色部分后添加以下三行信息： impor...

TensorFlow2.0！！！！Could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED错误解决方法

qq_40394402的博客

10-11

1577

解决**TensorFlow2.0**报错显存不足的问题Could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED错误解决方法 Could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED错误解决方法排查这个bug有两个步骤： 1、检测自己的cudnn版本，当出现报错的时候，直接查看jupyter notebook的运行黑色框。里面会有明确的报错信息，提示你这个版本的TensorFlow2要用哪一个

keras yolov3 Could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED

星之所望的博客

03-27

9692

训练、测试Tensorflow、Keras代码时，出现could not create cudnn handle: CUDNN_STATUS_NOT_INITIALIZED、error retrieving driver version: Unimplemented: kernel reported driver version not implemented on Windows、could n...

could not create cudnn handle: CUDNN_STATUS_NOT_INITIALIZED