...
--gres=gpu:1: Allocates 1 GPU to your job.--partition=partition_name: Specifies the partition to run the job in. Replacepartition_namewith the appropriate partition, likescu-gpu.--time=01:00:00: Requests 1 hour of runtime. Adjust the time as needed.--mem=8G: Requests 8 GB of memory.--cpus-per-task=4: Requests 4 CPU cores.--pty bash: Launches an interactive bash shell after resources are allocated.
SBATCH
| Code Block |
|---|
#!/bin/bash
#SBATCH --job-name=gpu_job # Job name
#SBATCH --output=output_file.txt # Output file
#SBATCH --partition=gpu_partition # Partition to run the job (e.g., scu-gpu)
#SBATCH --gres=gpu:1 # Request 1 GPU
#SBATCH --time=01:00:00 # Max runtime (1 hour)
#SBATCH --mem=8G # Memory requested
#SBATCH --cpus-per-task=4 # Number of CPU cores per task
# Your commands here
srun python my_script.py |