python - 关于使用 talib 与 gpu 或 tpu 使计算 EMA 和其他更快

Question

所以我正在用 ta-lib for python 做一个项目，我需要计算 EMA 和其他目标。数据是 10 年的历史数据，及时的库存数据总计为 1703，当我需要计算 EMA 仅针对 200 只股票时，它需要花费大约 19S~30S 来减慢，因为及时库存更新了 5s e，我希望它计算 10s更快，我想如果我可以使用 gpu 或 tpu 它会很快然后只使用 cpu 所以我的问题是 ta-lib 可以使用 gpu 还是 tpu ??? 以及如何使用它？如果无论如何不能让它快点？1703个股票？谢谢解答！！！

Answer 1

TA-Lib 不支持 GPU 或 TPU。您可以分叉它并在其中添加 GPU 支持到EMA 实现。使用修补的库重建和欺骗原始库。
或者您可以在自己的 lib 中使用 GPU（代码很简单）实现 EMA 计算并使用它而不是 TA-Lib。

顺便说一句，我有自己的 TA-Lib 分支，可以暂停并从保存状态继续。但它看起来对你的情况没有帮助，并且缺少 python 绑定。