Skip to content

Commit

Permalink
Create submit_iter.sh
Browse files Browse the repository at this point in the history
  • Loading branch information
colbyford committed Dec 6, 2024
1 parent bdd4a45 commit 3946859
Showing 1 changed file with 32 additions and 0 deletions.
32 changes: 32 additions & 0 deletions submit/multiple/submit_iter.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,32 @@
#!/bin/bash
#SBATCH --job-name="af3_iter"
#SBATCH --partition=GPU
#SBATCH --nodes=1
#SBATCH --ntasks-per-node=32
#SBATCH --mem=256GB
#SBATCH --gres=gpu:L40S:2
#SBATCH --time=1:00:00

module load singularity

export AF3_RESOURCES_DIR=/users/cford/alphafold3_resources

export AF3_IMAGE=${AF3_RESOURCES_DIR}/image/alphafold3.sif
export AF3_CODE_DIR=${AF3_RESOURCES_DIR}/code
export AF3_INPUT_DIR=${AF3_RESOURCES_DIR}/examples/${1}
export AF3_OUTPUT_DIR=${AF3_RESOURCES_DIR}/examples/${1}/output
export AF3_MODEL_PARAMETERS_DIR=${AF3_RESOURCES_DIR}/weights
export AF3_DATABASES_DIR=${AF3_RESOURCES_DIR}/databases

singularity exec \
--nv \
--bind $AF3_INPUT_DIR:/root/af_input \
--bind $AF3_OUTPUT_DIR:/root/af_output \
--bind $AF3_MODEL_PARAMETERS_DIR:/root/models \
--bind $AF3_DATABASES_DIR:/root/public_databases \
$AF3_IMAGE \
python ${AF3_CODE_DIR}/alphafold3/run_alphafold.py \
--json_path=/root/af_input/alphafold_input.json \
--model_dir=/root/models \
--db_dir=/root/public_databases \
--output_dir=/root/af_output

0 comments on commit 3946859

Please sign in to comment.