Try to use SLURM_STEP_GPUS for device list if CUDA_VISIBLE_DEVICES is not set#3577

copybara-service[bot] merged 1 commit intoAI-Hypercomputer:mainAI-Hypercomputer/maxtext:mainfrom

ROCm:gw_check_slurm_gpusROCm/maxtext:gw_check_slurm_gpusCopy head branch name to clipboard