Tensorflow需要花费大量时间来恢复ckpt模型

Question

在

tensorflow中恢复ckpt模型需要花费大量时间。我目前正在使用tensorflow-gpu 1.0.0.rc0。我的模型大约是500M，恢复模型大约需要1分钟。但是，当我使用 tensorflow-gpu 0.11.0 恢复相同的模型时，只需要几秒钟。知道为什么会这样吗？

Answer 1

很可能你的1.0.0rc0安装(为什么不更新到1.0？(使用的是你通过pip获得的纯python protobuf实现，而0.11安装使用的是谷歌改进的二进制发行版。

安装 protobuf 的优化版本，如 https://www.tensorflow.org/versions/master/get_started/os_setup#protobuf_library_related_issues 中所述但是您可能希望在 URL 中将 3.1.0 换成 3.2.0 以获得较新版本(我遇到了 3.2"更新"通过 pip 并替换了优化的 3.1 的问题(，所以而不是

pip install --upgrade https://storage.googleapis.com/tensorflow/linux/cpu/protobuf-3.1.0-cp27-none-linux_x86_64.whl

用

pip install --upgrade https://storage.googleapis.com/tensorflow/linux/cpu/protobuf-3.2.0-cp27-none-linux_x86_64.whl

或类似，具体取决于您的平台和 Python 版本。