CPU BiasOp only supports NHWC.

inoryy / reaver

Reaver: Modular Deep Reinforcement Learning Framework. Focused on StarCraft II. Supports Gym, Atari, and MuJoCo.

MIT License

554 stars 89 forks source link

CPU BiasOp only supports NHWC. #4

Closed mushroom1116 closed 6 years ago

mushroom1116 commented 6 years ago

I received some errors when running the code. InvalidArgumentError (see above for traceback): CPU BiasOp only supports NHWC. [[Node: Conv/BiasAdd = BiasAdd[T=DT_FLOAT, data_format="NCHW", _device="/job:localhost/replica:0/task:0/cpu:0"](Conv/convolution, Conv/biases/read)]] I want to use 2 gpu so I modified the args.gpu=2.

mushroom1116 commented 6 years ago

The reason is that my tensorflow doesn't support GPU. pip install tensorflow-gpu==1.3