p-doom/crowd-code-dataset-0.2 huggingface dataset:
[content] test.sh (shellscript)
      
#!/bin/bash
#SBATCH --nodes=1
#SBATCH --ntasks-per-node=4
#SBATCH --gres=gpu:4
#SBATCH --cpus-per-task=2
#SBATCH --time=00:01:00
#SBATCH --job-name=gpu_check
#SBATCH --output=gpu_check.out

# This command will be run for each of the 4 tasks
# Each task will print its unique environment
srun bash -c 'echo "Host: $(hostname), Slurm Task ID: $SLURM_TASK_PID, Local Rank: $SLURM_LOCALID, CUDA_VISIBLE_DEVICES: $CUDA_VISIBLE_DEVICES"'