Commit 37b17f17 authored by yuhai's avatar yuhai
Browse files

Update run_multimer.sh

parent d4c46da3
#!/bin/bash #!/bin/bash
export DTKROOT=/opt/dtk-22.04.2 export DTKROOT=/opt/dtk
export AMDGPU_TARGETS="gfx906" export AMDGPU_TARGETS="gfx928"
export ROCMVER=22.04.2 export ROCMVER=24.04.2
export DTK_HOME=/opt/dtk-22.04.2 export DTK_HOME=/opt/dtk
export ROCM_PATH=${DTK_HOME} export ROCM_PATH=${DTK_HOME}
export HIP_PATH=${DTK_HOME}/hip export HIP_PATH=${DTK_HOME}/hip
export PATH=${DTK_HOME}/bin:${DTK_HOME}/llvm/bin:${DTK_HOME}/hip/bin:${DTK_HOME}/miopen/bin:$PATH export PATH=${DTK_HOME}/bin:${DTK_HOME}/llvm/bin:${DTK_HOME}/hip/bin:${DTK_HOME}/miopen/bin:$PATH
...@@ -11,16 +11,16 @@ export INCLUDE=${DTK_HOME}/include:${DTK_HOME}/hip/include:${DTK_HOME}/llvm/incl ...@@ -11,16 +11,16 @@ export INCLUDE=${DTK_HOME}/include:${DTK_HOME}/hip/include:${DTK_HOME}/llvm/incl
export C_INCLUDE_PATH=${DTK_HOME}/include:${DTK_HOME}/hip/include:${DTK_HOME}/llvm/include:$C_INCLUDE_PATH export C_INCLUDE_PATH=${DTK_HOME}/include:${DTK_HOME}/hip/include:${DTK_HOME}/llvm/include:$C_INCLUDE_PATH
export CPLUS_INCLUDE_PATH=${DTK_HOME}/include:$CPLUS_INCLUDE_PATH export CPLUS_INCLUDE_PATH=${DTK_HOME}/include:$CPLUS_INCLUDE_PATH
export MIOPEN_SYSTEM_DB_PATH=${DTK_HOME}/miopen/share/miopen/db/ export MIOPEN_SYSTEM_DB_PATH=${DTK_HOME}/miopen/share/miopen/db/
source /root/env.sh source ./env.sh
cd /root/Uni-Fold-main cd /root/Uni-Fold
fasta_path=/root/Uni-Fold-main/data/H1036.fasta fasta_path=/root/Uni-Fold/data/H1036.fasta
input_dir_base=/root/Uni-Fold-main/data input_dir_base=/root/Uni-Fold/data
output_dir_base=/root/Uni-Fold-main/data output_dir_base=/root/Uni-Fold/data
database_dir=/alphafold/alphafold/ database_dir=/alphafold/
max_template_date=2020-05-01 max_template_date=2020-05-01
model_name=multimer_ft #model_2_ft model_name=multimer_ft #model_2_ft
param_path=/root/Uni-Fold-main/Alphafold/multimer.unifold.pt param_path=/root/Uni-Fold/Alphafold/multimer.unifold.pt
echo "Starting homogeneous searching..." echo "Starting homogeneous searching..."
python unifold/homo_search.py \ python unifold/homo_search.py \
...@@ -40,7 +40,7 @@ python unifold/homo_search.py \ ...@@ -40,7 +40,7 @@ python unifold/homo_search.py \
echo "Starting prediction..." echo "Starting prediction..."
fasta_file=$(basename $fasta_path) fasta_file=$(basename $fasta_path)
target_name=${fasta_file%.fa*} target_name=${fasta_file%.fa*}
export CUDA_VISIBLE_DEVICES=0,1,2,3
python3 unifold/inference.py \ python3 unifold/inference.py \
--model_name=$model_name \ --model_name=$model_name \
--param_path=$param_path \ --param_path=$param_path \
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment