chainer.backend.GpuDevice

class chainer.backend.GpuDevice(device)

Device for GPU (CuPy) backend

Methods

__enter__()

A dummy definition that simply raises RuntimeError.

chainer.using_device() should be used instead.

__exit__(exc_type, exc_value, traceback)

A dummy definition that should never be called.

create_context()

Returns a context manager in which the device is made current.

See also

chainer.using_device() calls this method internally.

static from_array(array)

static from_device_id(device_id)

Returns a GpuDevice corresponding to the CUDA device ID.

is_array_supported(array)

Returns if the specified array is compatible with the device. :param array: An array to be checked :type array: N-dimensional array

Returns

True if the array is compatible with the device. Otherwise False is returned.

send(arrays)

Transfers given arrays to the device.

Parameters

arrays – Array or arrays of NumPy, CuPy, or ChainerX.

Returns

Transferred arrays.

send_array(array)

use()

Makes the device current in the current thread.

__eq__(other)

Return self==value.

__ne__(other)

Return self!=value.

__lt__()

Return self<value.

__le__()

Return self<=value.

__gt__()

Return self>value.

__ge__()

Return self>=value.

Attributes

name

A unique name of the device.

supported_array_types = (<class 'chainer.backends.cuda.ndarray'>,)

xp = <object object>