31 lines
959 B
Bash
31 lines
959 B
Bash
#!/usr/bin/env bash
|
|
|
|
#SBATCH --job-name=cuda-singleblock
|
|
# Forschergruppe Schug
|
|
#SBATCH --account=hkf6
|
|
# 48 Cores, 512GiB RAM, 4x NVIDIA A100 GPUs
|
|
#SBATCH --partition=booster
|
|
# Right now we're using a single node
|
|
#SBATCH --nodes=1
|
|
# Number of MPI processes
|
|
#SBATCH --ntasks=1
|
|
# For now, we are using a single GPU only
|
|
#SBATCH --gres=gpu:1
|
|
#SBATCH --time=01:00:00
|
|
#SBATCH --output=logs/cuda-singleblock-%j.log
|
|
#SBATCH --error=logs/cuda-singleblock-%j.log
|
|
|
|
SOURCE_DIR=/p/project/cellsinsilico/paulslustigebude
|
|
OUTPUT_DIR=/p/scratch/cellsinsilico/paul/nastja-out/cuda-singleblock
|
|
|
|
mkdir -p "${OUTPUT_DIR}"
|
|
source "${SOURCE_DIR}/activate-nastja-modules"
|
|
|
|
# This is actually the default behavior for a single task anyways
|
|
# However I'm leaving this here for documentation reasons
|
|
export CUDA_VISIBLE_DEVICES=0
|
|
|
|
srun --unbuffered "${SOURCE_DIR}/nastja/build-cuda/nastja" \
|
|
-c "${SOURCE_DIR}/ma/experiments/configs/genesis-singleblock.json" \
|
|
-o "${OUTPUT_DIR}"
|