pub_soft/foldingdiff
3
0
Fork 0
mirror of https://github.com/microsoft/foldingdiff.git synced 2026-06-04 13:30:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

Redirect stdout

Browse Source
This commit is contained in:
Kevin Wu
2022-09-11 21:47:12 -07:00
parent 6e77dbaa48
commit b7b18f8be9
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
bin/omegafold_across_gpus.py
View File

@@ -48,8 +48,13 @@ def run_omegafold(input_fasta: str, outdir: str, gpu: int) -> str:
"""
logging.info(f"Running omegafold on {input_fasta} > {outdir} with gpu {gpu}")
assert shutil.which("omegafold")
cmd = f"omegafold {input_fasta} {outdir} --device cuda:{gpu}"
output = subprocess.call(cmd, shell=True)
cmd = f"CUDA_VISIBLE_DEVICES={gpu} omegafold {input_fasta} {outdir} --device cuda:0"
bname = os.path.splitext(os.path.basename(input_fasta))[0]
with open(
os.path.join(outdir, f"omegafold_{bname}_gpu_{gpu}.stdout"), "wb"
) as sink:
output = subprocess.call(cmd, shell=True, stdout=sink)
def build_parser() -> argparse.ArgumentParser: