Get some errors when tested on Windows 10
Closed this issue · 0 comments
woaipichuli commented
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?