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 

或相似。