Add single block configs
This commit is contained in:
@@ -0,0 +1,31 @@
|
||||
#!/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
|
||||
#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-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 "${SOURCE_DIR}/nastja/build-cuda/nastja" \
|
||||
-c "${SOURCE_DIR}/ma/experiments/configs/genesis-singleblock.json" \
|
||||
-o "${OUTPUT_DIR}"
|
||||
Reference in New Issue
Block a user