使用intel-tensorflow

起因

原本使用的是tensorflow-1.15.0，但是会报错：

Executor failed to create kernel. Invalid argument: Default MaxPoolingOp only supports NHWC on device type CPU

修改数据格式可以解决，但是要该的地方很多，太麻烦了。

intel-tensorflow 对 tensorflow 进行了优化，具体内容可参考官网提升 TensorFlow* 在英特尔® 至强® 处理器上的推理性能

比如这个地方，可以不用改代码，将 tensorflow 替换为 intel-tensorflow 就行。

步骤

1. 卸载原有的tensorflow

pip uninstall tenforflow

2. 安装新的

pip install intel-tensorflow==1.15

* 这里pip安装的时候出了一点问题，会报错：

ERROR: intel-tensorflow has an invalid wheel, multiple .dist-info directories found: intel_tensorflow-1.15.0.dist-info, tensorflow-1.15.0.dist-info

我的pip版本：21.x

查了下，这属于pip的bug（我还以为是我自己没卸干净），pip并没有在维护intel-tensorflow。作者说这是20.x的bug，19.x及以前的版本不会有这个问题，以后的版本会修复（然而我的21.x还是存在这个问题）。

那就只能回退pip了

pip install -U pip==19

再重新下载intel-tensorflow就不会报错了。

【总结】pip相关的文章