cuda.bindings.nvml.system_get_topology_gpu_set

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

Retrieve the set of GPUs that have a CPU affinity with the given CPU number

Parameters:

cpuNumber (unsigned int) – The CPU number

Returns:

An array of device handles for GPUs found with affinity to cpuNumber

Return type:

array