InvalidArgumentError (see above for traceback): Default MaxPoolingOp only supports NHWC on device ty

对于这个问题,有不同的解决方式;
InvalidArgumentError (see above for traceback): Default MaxPoolingOp only supports NHWC on device type CPU
[[Node: ssd_300_vgg/pool1/MaxPool = MaxPoolT=DT_FLOAT, data_format=“NCHW”, ksize=[1, 1, 2, 2], padding=“SAME”, strides=[1, 1, 2, 2], _device="/job:localhost/replica:0/task:0/device:CPU:0"]]

  • 解决方法1:将代码中的 ‘NCHW’ as ‘NHWC’
    因为对于NCHW是支持GPU的,而NHWC是支持CPU的,所以如果你的电脑没有相应的GPU硬件或者训练的时候不需要用到GPU,那么你就可以将NCHW 改成 NHWC然后把有关GPU训练的相关设置都关闭或者取消

  • 解决方法2:如果电脑是有GPU而同时你也想用GPU对自己的模型进行训练,那么就将这个地方设置成NCHW,但是即使这样也不一定就能保证问题解决,因为导致这个错误的一个很大的原因在于:那绝大多数是因为你的GPU相关设置没有匹配好。如果你有GPU硬件而且你将这个选项设置成NCHW但是依然报这个错误,,举个例子来说:开始的时候我用了tensorflow-gpu=1.10.0,cudnn=7.6.5,cudatoolkit=10.2他还是报错,同样的错误,但是当我将这些东西全部卸载,换成了tensorflow-gpu=1.11.0,cudnn=7.6.5,cudatoolkit=9.2然后这个问题就解决了。

综上所述,如果你遇到了这个问题,不要简单地将data_format="NCHW"和 data_format="NHWC"之间简单的切换,要注重环境配置方面的问题。

希望你也能够不受环境的困扰。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

暖仔会飞

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值