title: error while running tensorflow on P100
body: <p>I am trying to run tf_cnn_benchmarks on P100 GPU. I am using TF-1.60.</p>
  
  <p>I am getting the following message:</p>
  
  <p>2018-03-23 11:02:22.043275: E tensorflow/core/common_runtime/executor.cc:645] Executor failed to create kernel. Invalid argument: Default MaxPoolingOp only supports NHWC on device type CPU
       [[Node: v/tower_0/cg/mpool0/MaxPool = MaxPoolT=DT_FLOAT, data_format="NCHW", ksize=[1, 1, 3, 3], padding="SAME", strides=[1, 1, 2, 2], _device="/job:localhost/replica:0/task:0/device:CPU:0"]]
  I0323 11:02:22.115472 47942823427200 tf_logging.py:116] Error reported to Coordinator: , Default MaxPoolingOp only supports NHWC on device type CPU
       [[Node: v/tower_0/cg/mpool0/MaxPool = MaxPoolT=DT_FLOAT, data_format="NCHW", ksize=[1, 1, 3, 3], padding="SAME", strides=[1, 1, 2, 2], _device="/job:localhost/replica:0/task:0/device:CPU:0"]]</p>
  
  <p>I am confused. Why it is asking for stuff for CPU. I am using the following command:</p>
  
  <p>srun python tf_cnn_benchmarks.py --num_gpus=1 --batch_size=32 --model=resnet50 --variable_update=parameter_server</p>
  
  <p>Thanks</p>