33 lines
982 B
Bash
33 lines
982 B
Bash
#!/usr/bin/env bash
|
|
|
|
#SBATCH --job-name=cuda
|
|
# 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
|
|
# TODO: Change the config and set to this the maximum of 48
|
|
#SBATCH --ntasks-per-node=16
|
|
#SBATCH --cpus-per-task=1
|
|
# For now, we are using a single GPU only
|
|
#SBATCH --gres=gpu:1
|
|
#SBATCH --time=01:00:00
|
|
#SBATCH --output=logs/cuda-%j.log
|
|
#SBATCH --error=logs/cuda-%j.log
|
|
|
|
SOURCE_DIR=/p/project/cellsinsilico/paulslustigebude
|
|
OUTPUT_DIR=/p/scratch/cellsinsilico/paul/nastja-out/cuda
|
|
|
|
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 "${SOURCE_DIR}/nastja/build-cuda/nastja" \
|
|
-c "${SOURCE_DIR}/ma/experiments/configs/genesis.json" \
|
|
-o "${OUTPUT_DIR}"
|