search

HeliXonProtein / OmegaFold

OmegaFold Release Code
Apache License 2.0
532 stars 75 forks source link

device option #42

Open rmarabini opened 1 year ago

rmarabini commented 1 year ago

Hi,

Could you help me to understand how the option "device" works?
Should OmegaFold run on GPU if I provide the parameter "--device cuda"? How can I choose a particular GPU?

Right now Omegafold is running in my workstation using 32 cores and ignoring the GPUs.

thanks in advance
rpkarandev commented 1 year ago

Yes, I could confirm that. OmegaFold used GPU with explicit use of "--device cuda". "cuda" in lowercase

Please make sure the python environment has the necessary Cuda library installed.