广告:或许有人搞频域攻击的对抗水印吗?可以交流一下。最近在学一个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。成功!!