cuda.bindings.nvml.system_get_topology_gpu_set

cuda.bindings.nvml.system_get_topology_gpu_set(unsigned int cpu_number)

Retrieve the set of GPUs that have a CPU affinity with the given CPU number For all products. Supported on Linux only.

Parameters:

cpu_number (unsigned int) – The CPU number.

Returns:

An array of device handles for GPUs found with

affinity to cpu_number.

Return type:

intptr_t

See also

nvmlSystemGetTopologyGpuSet