当前位置: 代码迷 >> 综合 >> 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 ty

热度:65   发布时间:2023-12-10 07:54:16.0

转,想用compact CNN训练开源数据,在只装了tensorflow cpu版本的pycharm上跑,报该错,上网查的结果:

  1. 一般是由于模型是用GPU训练出来的,数据格式为NCHW,但是你用的tensorflow-cpu版本,仅支持NHWC格式。所以安装tensorflow-gpu版本就行。
  2. 如果安了tensorflow-gpu版本还是失败。看下运行log是否使用GPU。如下表示未使用到GPU,还是用的CPU在跑。xiang
    failed call to cuInit: CUDA_ERROR_NO_DEVICE
  相关解决方案