當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文

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。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。