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

Question

在张量流中恢复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安装使用来自google的改进二进制分发。

按照https://www.tensorflow.org/versions/master/get_started/os_setup#protobuf_library_related_issues的详细说明安装优化版本的protobuf 但你可能想在URL中交换3.1.0 for 3.2.0以获得更新的版本（我遇到的问题是3.2＆＃34;更新＆＃34;来自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版本。