Table Of Contents
Table Of Contents

mxnet.context¶

Context management API of mxnet.

cpu([device_id])

Returns a CPU context.

cpu_pinned([device_id])

Returns a CPU pinned memory context.

current_context()

Returns the current context.

gpu([device_id])

Returns a GPU context.

num_gpus()

Query CUDA for the number of GPUs present.

Context(device_type[, device_id])

Constructs a context.