导入trax需要很长时间才能加载

Question

我第一次加载这个图书馆的时候就被困住了。在我的本地计算机中，我至少需要40多岁才能在本地的木星笔记本上加载trax，在共享的Colab环境下加载它需要超过1分钟。

import trax

我不确定这是我安装的问题，还是我正在使用的特拉克斯版本中的BUG？

我在trax是新来的，事实上，我的经验是使用Keras和TensorFlow，所以我想从trax社区的某个人那里得到一些意见，不管这是否正常。

提前谢谢！

BTW：我使用的是trax 1.4.1和Python 3.9.6，我的本地计算机有以下规格：

 Intel(R) Core(TM) i5-6200U CPU @ 2.30GHz, 4 cores and 16GB RAM. 

Answer 1

经过一段时间的搜索，这似乎是正常的行为。事实上，官方的Trax存储库中提出了一个关于这一点的问题：导入trax需要17秒#1368。

显然，包含用于数学操作的trax重新实现的fastmath模块有很多依赖项。

来自问题线程的

这是一个众所周知的问题发生在基本上所有的设置(本地，colab，gpu集群)，它不是一个大的问题，运行长期的实验，但它确实使本地调试困难。我试过用分析器调试导入图，但没有成功。看起来就像即使从trax导入快速数学也有很多依赖关系这里是由导入库为trax.fastmath.init模块生成的树: import has - init.py out：https://gist.github.com/syzymon/3bb6f59063f918b4b62b77cdb223da72

总之，无论你用17秒还是40秒，就像我一样，在Trax中是一种众所周知的行为。