search

sokrypton / ColabDesign

Making Protein Design accessible to all via Google Colab!
Other
575 stars 137 forks source link

How to use gpu or cuda for design binder #109

Open cloudaner115 opened 1 year ago

cloudaner115 commented 1 year ago

I think the current design uses cpu, it feels too slow, how to use gpu

shanilpanara commented 1 year ago

Hi @404479768, you can ask for a GPU from Google Colab: Menu -> Runtime -> Change Runtime Type -> Hardware Accelerator: GPU

To check if you have a GPU, you can run the bash command !nvidia-smi, with the exclamation mark at the front. If you get an error message, it means you haven't been assigned a GPU.

Like so:

image

If the design is still very slow, this could be due to a number of other factors, e.g. the size of your protein.