remote/bin/slurm-shell @ 4f28fbfd7d63

More
author Steve Losh <steve@stevelosh.com>
date Tue, 09 Apr 2024 09:20:04 -0400
parents 417f91fae3c2
children (none)
#!/usr/bin/env bash

set -euo pipefail

read -p "Account? "
acct="$REPLY"

read -p "Hours? "
hours="$REPLY"

read -p "CPUs? "
cpus="$REPLY"

read -p "Memory (gb)? "
mem_gb="$REPLY"
mem_kb=$(( mem_gb * 1024 * 1024 / cpus ))

exec salloc --account="$acct" --nodes=1 --ntasks-per-node=1 --mem-per-cpu="$mem_kb"K --cpus-per-task="$cpus" --time="$hours":00:00