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的改进的二进制分发版。

安装的protobuf的优化版本上https://www.tensorflow.org/versions/master/get_started/os_setup#protobuf_library_related_issues 详见但你可能要交换3.1.0为3.2.0的网址，以获取新的版本（我有问题的是，3.2“更新”通过PIP来到并根据您的平台和Python版本上换下的优化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 

或相似。