gpu - 有没有办法在多 GPU 上运行 Rapids 时间序列模块（ARIMA、ESM）？

Question

我有一个带有 2 个 Tesla P100 gpu 的节点。

当我运行 rapids.tsa.ARIMA（或 ESM）时，它只会使用其中一个 GPU。有没有办法利用多 GPU 来训练模型？就像在 rapids-dask-xgboost 中一样？

Answer 1

不是根据 cuml文档。对于多节点多 GPU，请检查表上的注释列。

Answer 2

时间序列模型、Arima 和 Holtwinters 的多节点多 gpu 实现不存在。我建议在 cuML 的 github 存储库 ( https://github.com/rapidsai/cuml ) 上创建一个功能请求。