Genentech / equifold

Official code repository for EquiFold: Protein Structure Prediction with a Novel Coarse-Grained Structure Representation

Home Page:https://www.biorxiv.org/content/10.1101/2022.10.07.511322v1

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

how to use gpu

bioermaf opened this issue · comments

how to edite the script to use gpu

$ python run_inference.py --model science --model_dir models --seqs tests/data/inference_science_input.csv --ncpu 1 --out_dir out_tests

if you are on a gpu-equipped platform and currently installed the tool, the script is written to run the inference on gpu.