Tensorflow-MKL在3D数据上显示错误

Question

我正在训练一个模型来执行体积分割（3D数据）。由于输入数据的大小，我在CPU上进行培训。我正在使用带有tensorflow-mkl和keras的anaconda环境。训练模型时，我得到了error：

"tensorflow.python.framework.errors_impl.InvalidArgumentError: Value for attr 'data_format' of "NDHWC" is not in the list of allowed values: "NHWC", "NCHW"

但是，在他们的github上，它说它适用于体积分割（https://github.com/intel/mkl-dnn）。如何解决此问题，以便可以使用mkl训练3D-Unet？

Answer 1

尝试将tensorflow更新到1.8.0版本。 希望这能解决您的问题。