time-series - 使用 DeepAR GPU

Question

我正在使用 Gluonts 构建 DeepAR 模型，但即使我使用 cox = 'gpu' 但会引发错误，也需要花费大量时间来运行训练对象。我的机器有 GPU，但该选项不起作用。任何帮助深表感谢...

Answer 1

您可以检查您的 mxnet 当前版本，我相信您使用的是 CPU 版本。

请检查以下内容：

import mxnet as mx
    print(f'mxnet version: {mx.__version__}')
    print(f'Number of GPUs: {mx.context.num_gpus()}')

它应该返回 GPU 的数量