当前位置: 首页>>代码示例 >>用法及示例精选 >>正文

Python mxnet.context.gpu用法及代码示例


用法:

mxnet.context.gpu(device_id=0)

参数：

device_id：(int, optional) - 设备的设备 ID，GPU 需要。

返回：

context：- 对应的 GPU 上下文。

返回类型：

mxnet.context.Context

返回一个 GPU 上下文。

此函数是 Context(‘gpu’, device_id) 的快捷方式。一个节点上的 K 个 GPU 通常编号为 0,…,K-1。

例子：

>>> cpu_array = mx.nd.ones((2, 3))
>>> cpu_array.context
cpu(0)
>>> with mx.gpu(1):
...     gpu_array = mx.nd.ones((2, 3))
>>> gpu_array.context
gpu(1)
>>> gpu_array = mx.nd.ones((2, 3), ctx=mx.gpu(1))
>>> gpu_array.context
gpu(1)

相关用法


注：本文由纯净天空筛选整理自apache.org大神的英文原创作品 mxnet.context.gpu。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。