launch_with_binding.sh

#!/bin/bash

export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7
"${@:2}"
# LOCAL_RANK=$1
# shift

# numa_map=(0 1 2 3 4 5 6 7)
# NUMA_ID=${numa_map[$LOCAL_RANK]}
# numactl --cpunodebind=${NUMA_ID} --membind=${NUMA_ID} "$@"