Closed woaipichuli closed 6 years ago
When I test the code on windows 10, I have get some errors. Could you please help me out? I use tensorflow-1.7 python-3.5 GPU 1080 Ti
InvalidArgumentError (see above for traceback): Conv2DCustomBackpropFilterOp only supports NHWC. [[Node: train_op/gradients/Conv/Conv2D_grad/Conv2DBackpropFilter = Conv2DBackpropFilter[T=DT_FLOAT, data_format="NCHW", dilations=[1, 1, 1, 1], padding="SAME", strides=[1, 1, 1, 1], use_cudnn_on_gpu=true, _device="/job:localhost/replica:0/task:0/device:CPU:0"](transpose/_1297, train_op/gradients/Conv/Conv2D_grad/Const, train_op/gradients/Conv/BiasAdd_grad/tuple/control_dependency/_1299)]]
I referred to the previous issues. I found it run correctly if I add --nhwc. But I want to run it on GPU, what should I do?
InvalidArgumentError (see above for traceback): Conv2DCustomBackpropFilterOp only supports NHWC. [[Node: train_op/gradients/Conv/Conv2D_grad/Conv2DBackpropFilter = Conv2DBackpropFilter[T=DT_FLOAT, data_format="NCHW", dilations=[1, 1, 1, 1], padding="SAME", strides=[1, 1, 1, 1], use_cudnn_on_gpu=true, _device="/job:localhost/replica:0/task:0/device:CPU:0"](transpose/_1297, train_op/gradients/Conv/Conv2D_grad/Const, train_op/gradients/Conv/BiasAdd_grad/tuple/control_dependency/_1299)]]
I referred to the previous issues. I found it run correctly if I add --nhwc. But I want to run it on GPU, what should I do?