a47f9aaf01a0b56a597fef6798a3741c69cf89a0
examples/rosettafold_all_atom.md
... | ... | @@ -0,0 +1,42 @@ |
1 | +#!/usr/bin/env bash |
|
2 | + |
|
3 | +## SBGrid RosettaFold-All-Atom example |
|
4 | +## |
|
5 | +## James Vincent - help@sbgrid.org |
|
6 | +## May 13, 2024 |
|
7 | + |
|
8 | + |
|
9 | +# SLURM |
|
10 | +#SBATCH --mem=64G |
|
11 | +#SBATCH -t 0:30:00 |
|
12 | +#SBATCH -p gpu_quad |
|
13 | +#SBATCH --gres=gpu:1 |
|
14 | + |
|
15 | + |
|
16 | +# cd /temp_work/ch199734/114504_rosettafoldaa # BCH E3 |
|
17 | +cd /home/jjv16/tickets/114504_rosettafoldaa # O2 |
|
18 | + |
|
19 | +module load cuda/11.2 |
|
20 | + |
|
21 | +# Start SBGrid environment |
|
22 | +source /programs/sbgrid.shrc |
|
23 | + |
|
24 | +# Set version of RFAA |
|
25 | +export ROSETTAFOLDAA_X=bf21483 |
|
26 | + |
|
27 | +# Set critical env variables |
|
28 | +export DGLBACKEND=pytorch |
|
29 | +#export ROSETTAFOLD_DB_PATH=/programs/local/biogrids/rosettafold/ # BCH E3 |
|
30 | +export ROSETTAFOLD_DB_PATH=/n/scratch/users/j//jjv16/rosettafold # O2 |
|
31 | +export MY_RFAA_DIR=/programs/x86_64-linux/rosettafoldaa/${ROSETTAFOLDAA_X}/RoseTTAFold-All-Atom |
|
32 | + |
|
33 | +# Call the correct python bundled with RFAA |
|
34 | +python.rfaa --version |
|
35 | + |
|
36 | +# Run inference, override hydra parameters from YAML file |
|
37 | +python.rfaa -m rf2aa.run_inference \ |
|
38 | + --config-name sbgrid_example -cd ./ \ |
|
39 | + checkpoint_path=${MY_RFAA_DIR}/RFAA_paper_weights.pt \ |
|
40 | + database_params.command=${MY_RFAA_DIR}/make_msa.sh \ |
|
41 | + database_params.hhdb=${ROSETTAFOLD_DB_PATH}/pdb100_2021Mar03/pdb100_2021Mar03 |
|
42 | + |