cuDNN launch failure

报错：

InternalError (see above for traceback): cuDNN launch failure : input shape ([64,112,112,1])[[Node: depth_1/BatchNorm/FusedBatchNorm = FusedBatchNorm[T=DT_FLOAT, data_format="NHWC", epsilon=2e-05, is_training=true, _device="/job:localhost/replica:0/task:0/device:GPU:0"](_arg_depth_0_1/_265, ir_1/BatchNorm/Const, depth/BatchNorm/beta/read, color_1/Conv2d_0/conv/BatchNorm/Const_1, color_1/Conv2d_0/conv/BatchNorm/Const_1)]][[Node: orbbec_ir/accuracy/Mean/_359 = _Recv[client_terminated=false, recv_device="/job:localhost/replica:0/task:0/device:CPU:0", send_device="/job:localhost/replica:0/task:0/device:GPU:0", send_device_incarnation=1, tensor_name="edge_2955_orbbec_ir/accuracy/Mean", tensor_type=DT_HALF, _device="/job:localhost/replica:0/task:0/device:CPU:0"]()]]

原因是因为显存不够，在你得到程序最上面加上代码：

import os
os.environ['TF_FORCE_GPU_ALLOW_GROWTH'] = 'true'

问题解决。
后来运行程序又遇到这个问题，谷歌了半天，参考GITHUB上看到的方法，后来把

os.environ['TF_FORCE_GPU_ALLOW_GROWTH'] = 'true'

注掉就又可以运行了，太迷惑了，想不明白是为啥，只想静静。。。

cuDNN launch failure相关推荐

caffe不支持relu6_caffe cudaSuccess (4 vs. 0) unspecified launch failure
执行sudo /home/test/caffe-master/build/tools/caffe train --solver=/media/test/810a7433-6292-4fef-8e87- ...
使用Amazon Deep Learning AMI 快速实现 CUDA，cuDNN 和深度学习框架版本兼容
前言在开展深度学习项目时,我们通常会选择合适的深度学习框架.使用深度学习框架进行模型开发,能减少大量的重复代码工作.目前最流行的深度学习框架有:TensorFlow,PyTorch,MXNect,C ...
使用 Amazon Deep Learning AMI 快速实现 CUDA，cuDNN 和深度学习框架版本兼容
在开展深度学习项目时,我们通常会选择合适的深度学习框架.使用深度学习框架进行模型开发,能减少大量的重复代码工作.目前最流行的深度学习框架有:TensorFLow,PyTorch,MXNect,Caff ...
深度卷积生成对抗网络（DCGAN）原理与实现（采用Tensorflow2.x）
深度卷积生成对抗网络(DCGAN)原理与实现(采用Tensorflow2.x) GAN直观理解 DCGAN网络结构 GAN训练目标 DCGAN实现数据加载网络鉴别网络生成网络网络训练定义损 ...
cuSPARSE库：（十七）cusparseStatus_t 返回信息
(1)cusparseStatus_t 结构体的形式 typedef enum{CUSPARSE_STATUS_SUCCESS=0,CUSPARSE_STATUS_NOT_INITIALIZED=1 ...
Android AMS(android 11)
AMS庞大而复杂,我们今天就看下如何启动一个activity吧.只看流程. 可以结合这个图 https://blog.csdn.net/qq_15255121/article/details/1192 ...
【安卓 R 源码】Activity 启动流程及其生命周期源码分析
1. Activty 的生命周期 activity的生命周期 oncreate()->onstart()->onResume()->onPause()->onStop()-&g ...
Launcher 的启动
1.Launcher概述 Android系统启动的最后一步是启动一个Home应用程序,这个应用程序用来显示系统中已经安装的应用程序,这个Home应用程序就叫做Launcher.应用程序Launcher ...
ci/cd自动化测试_自动化CI / CD并花费更多时间编写代码
ci/cd自动化测试 This article was sponsored by Microsoft Visual Studio App Center. Thank you for supportin ...

cuDNN launch failure

cuDNN launch failure相关推荐

最新文章

热门文章