什么可能导致张量流导入如此缓慢?

时间:2018-03-01 15:45:33

标签: python performance tensorflow import

我已经使用了tensorflow很长一段时间了。最近我的脚本运行速度非常慢(> 80秒),早期(<1秒)。我将问题缩小到import tensorflow,这一直是一直在进行的(所有其他的lib和操作都在运行&lt;&lt; 1秒)。

我可能有一点痕迹,但我不知道如何处理它:当我在80秒导入期间键盘中断(Strg + C)执行时,通常会出现这种情况:

Traceback (most recent call last):
  File "/.../py_env/tf_unet/lib/python3.5/site.py", line 703, in <module>
    main()
  File "/.../py_env/tf_unet/lib/python3.5/site.py", line 694, in main
    execsitecustomize()
  File "/.../py_env/tf_unet/lib/python3.5/site.py", line 548, in execsitecustomize
    import sitecustomize
  File "/usr/lib/python3.5/sitecustomize.py", line 3, in <module>
    import apport_python_hook
  File "<frozen importlib._bootstrap>", line 969, in _find_and_load
  File "<frozen importlib._bootstrap>", line 954, in _find_and_load_unlocked
  File "<frozen importlib._bootstrap>", line 896, in _find_spec
  File "<frozen importlib._bootstrap_external>", line 1139, in find_spec
  File "<frozen importlib._bootstrap_external>", line 1113, in _get_spec
  File "<frozen importlib._bootstrap_external>", line 1225, in find_spec
  File "<frozen importlib._bootstrap_external>", line 1264, in _fill_cache
KeyboardInterrupt

这是否意味着什么?&#34;填充缓存&#34; (_fill_cache)错了?谁有这方面的体验?我能以某种方式解决这个问题吗?

到目前为止我已尝试过:

我将import tensorflow分解为我需要的模块(from tensorflow import train / python_io / compat),没有任何改进。

我发现其他人抱怨import tensorflow速度here herecorresponding SO question以及Output results here,但在&lt; {3}}范围内10秒并参考特定模块(tf.contribtf.learn)。所以从那里学到的东西并不多。我也在使用tensorflow 1.4.0,这显然解决了那里描述的问题。

仅供参考,我使用这段小代码来确定速度:

from timeit import default_timer as timer
print('import tensorflow')
start = timer()
import tensorflow
end = timer()
print('Elapsed time: ' + str(end - start))

1 个答案:

答案 0 :(得分:1)

这可能不是造成这种情况的唯一原因,但根据我的经验肯定会发挥作用。由于我在网络驱动器上安装了TF虚拟环境,导入Tensorflow时出现了严重的缓慢。在这方面,将虚拟环境移动到本地硬盘有很大帮助。

你可以尝试做一些适用于你环境的类似事情。