gluonts.mx.context 模块#

class gluonts.mx.context.MXContext[源文件]#

基类: object

定义自定义数据类型验证。

用于 Context 数据类型，参考 https://docs.pydantic.org.cn/#custom-data-types。

classmethod validate(v: Union[str, mxnet.context.Context]) → mxnet.context.Context[源文件]#

gluonts.mx.context.check_gpu_support() → bool[源文件]#

输出一条日志，并返回一个布尔值，指示当前安装的 MXNet 版本是否支持 GPU。

gluonts.mx.context.get_mxnet_context(gpu_number=0) → mxnet.context.Context[源文件]#

返回 CPU 或 GPU 上下文。

gluonts.mx.context.num_gpus(refresh=False)[源文件]#