cuda.bindings.nvml.unit_get_devices

cuda.bindings.nvml.unit_get_devices(intptr_t unit)

Retrieves the set of GPU devices that are attached to the specified unit.

Parameters:

unit (Unit) – The identifier of the target unit.

Returns:

An array of device handles for GPUs attached to the unit.

Return type:

array