#!/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} "$@"