1. 程式人生 > >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

  1. 一般是由於模型是用GPU訓練出來的,資料格式為NCHW,但是你用的tensorflow-cpu版本,僅支援NHWC格式。所以安裝tensorflow-gpu版本就行。
  2. 如果安了tensorflow-gpu版本還是失敗。看下執行log是否使用GPU。如下表示未使用到GPU,還是用的CPU在跑。
    failed call to cuInit: CUDA_ERROR_NO_DEVICE

     

  3. 看下程式碼中是否有如下程式碼:

    os.environ["CUDA_VISIBLE_DEVICES"] = "1"

    表示你使用的GPU裝置為1,可以改為0試試。