I. `sr` — Custom SLURM Interactive Job Launcher#

Overview: sr is a small wrapper script around srun that lets you launch an interactive GPU session on an HPC cluster with a short, memorable command — optionally targeting a specific node.

1. Write the Script#

Create a file named sr with the following content:

1
#!/bin/bash
2

3
# ===== Argument check =====
4
if [ -z "$1" ]; then
5
    echo "Usage:"
6
    echo "  sr <gpu_type> [node_id]"
7
    echo ""
8
    echo "Examples:"
9
    echo "  sr h100"
10
    echo "  sr h100 007"
11
    echo "  sr a100"
12
    exit 1
13
fi
14

15
GPU_TYPE="$1"
16
PARTITION="gpucompute-$GPU_TYPE"
17
NODE_ARG=""
18

19
# ===== Optional node targeting =====
20
if [ -n "$2" ]; then
21
    NODE_ARG="--nodelist=ghpc$2"
22
fi
23

24
# ===== Launch interactive session =====
25
srun \
26
  --gpus-per-node=1 \
27
  --cpus-per-gpu=4 \
28
  $NODE_ARG \
29
  --partition=$PARTITION \
30
  --time=12:00:00 \
31
  --pty /bin/bash

2. Install the Script#

Move the script to ~/bin and make it executable:

1
mkdir -p ~/bin
2
mv sr ~/bin/
3
chmod +x ~/bin/sr

Ensure ~/bin is on your PATH:

1
echo 'export PATH="$HOME/bin:$PATH"' >> ~/.bashrc
2
source ~/.bashrc

3. Usage#

1
sr h100          # Request any H100 node
2
sr h100 007      # Request H100 node ghpc007 specifically
3
sr a100          # Request any A100 node

💡 One-line Takeaway
Drop sr into ~/bin, add ~/bin to your PATH, and replace verbose srun commands with sr h100 or sr h100 007 to spin up an interactive GPU session instantly.

I. sr — Custom SLURM Interactive Job Launcher#

1. Write the Script#

2. Install the Script#

3. Usage#

I. `sr` — Custom SLURM Interactive Job Launcher#