E tensorflow/stream_executor/cuda/cuda_dnn.cc:82] CUDNN_STATUS_EXECUTION_FAILED

广告:或许有人搞频域攻击的对抗水印吗?可以交流一下。最近在学一个FAWA的方法,github上有,如果有人在跑这个项目可以跟我唠唠~need help~~~~~~~~

tf-gpu=1.12.0
cuda 9.0
cudnn7.1.4.18
keras=2.2.4
torch=1.1.0
torchvision=0.3.0
在这里插入图片描述
下面是问题截图
在这里插入图片描述
我的这个问题是经常出现的问题,但是有时候又不会出现这个问题,很烦!
解决办法试了很多,最后一个有用,其他都是试了但是没有效果。

方法1

加入下面三行,没用

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

方法2

禁用cudnn,网上都是用的pytorch,没用,我用的 tf 跟torch禁用没有关系

方法3

升级cudnn至7.1.4.18,原来是7.1.2,升级了也没变。但是我没改回去。

方法4(有效,但是我将cudnn改了,不知道是不是叠加生效,能跑就行!! )沙毕。没用了,服了。。

打开shell的sh文件,将batch_size=100改成50。成功!!
E tensorflow/stream_executor/cuda/cuda_dnn.cc:82] CUDNN_STATUS_EXECUTION_FAILED_第1张图片

你可能感兴趣的:(水印代码,tensorflow,dnn,深度学习)