HeliXonProtein/OmegaFold

device option

rmarabini opened this issue · 1 comments

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

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.