问使用DeepAR图形处理器
EN

Stack Overflow用户

提问于 2021-10-25 12:09:11

回答 1查看 31关注 0票数 1

我使用Gluonts来构建DeepAR模型，但是即使我使用cox = 'gpu‘运行训练对象也要花费很多时间，但是抛出了一个错误。我的机器有GPU，但选项不起作用。任何帮助都是非常感谢的。

发布于 2021-10-25 12:20:19

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

请检查以下各项：

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

它应该返回gpus的数量

票数 1

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/69707809

复制

相似问题

问使用DeepAR图形处理器EN