#!/usr/bin/env bash lrank=$OMPI_COMM_WORLD_LOCAL_RANK numa1=$((lrank)) netdev=mlx5_${lrank}:1 export CUDA_VISIBLE_DEVICES=$OMPI_COMM_WORLD_LOCAL_RANK export UCX_NET_DEVICES=${netdev} BINDING="--interleave=$numa1" echo "$(hostname) - $lrank device=$CUDA_VISIBLE_DEVICES binding=$BINDING" numactl ${BINDING} "$@"