nvidia-smi topo -m

def set_cpu_affinity(rank, num_cores_per_gpu=4, start_core_index=0):
    num_cores = psutil.cpu_count(logical=True)
    core_ids = list(range(num_cores))
    # Вычислить индекс начального и конечного ядер
    start_core = start_core_index + rank * num_cores_per_gpu
    end_core = start_core + num_cores_per_gpu
    # Получить список ядер ЦП, которые необходимо привязать
    cpu_affinity = core_ids[start_core:end_core]
    # Установите привязку ядра ЦП текущего процесса
    p = psutil.Process(os.getpid())
    p.cpu_affinity(cpu_affinity)
    print(f">> GPU {rank} is bound to CPU cores {cpu_affinity}")

set_cpu_affinity(rank)

htop