#!/bin/bash

srun -K \
  --container-image=PATH/TO/SLURM/IMAGE \
  --container-workdir="$(pwd)" \
  --container-mounts=/ALL/IMPORTANT/MOUNTS,"$(pwd)":"$(pwd)" \
  --partition=RTXA6000,RTX3090,A100-40GB,A100-80GB,H100,H200 \
  --job-name="python" \
  --nodes=1 \
  --gpus=1 \
  --ntasks=1 \
  --cpus-per-task=24 \
  --mem=64G \
  --time=1-0 \
  --export="NLTK_DATA=/PATH/TO/NLTK_DATA" \
  python3 "$@"