search

generatebio / chroma

A generative model for programmable protein design
Apache License 2.0
696 stars 90 forks source link

problem about "protein = chroma.design(protein)" #34

Closed ning-2022 closed 10 months ago

ning-2022 commented 11 months ago

Hi Chroma experts,

Thanks for your distinguished program for protein design. I'm trying to use it on my laptop that having one GPU and one CPU. When I tried to use this command from tutorial in the python3 interface, like that " protein = chroma.design(protein) " , some errors were accidentally appeared.

RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!

Please help me, I also am a freshman in Python programming.

Best regards, Ning

ning-2022 commented 11 months ago

Hi, I have a quick feedback that if I modifies the input path like this " protein = Protein('/home/neil/sample.cif', device='cuda:0') ", this kind of problem seems to be solved. But I'm not sure this way is correct. Any suggestions are welcome.

Ning

wujiewang commented 11 months ago

That sounds like the right fix, Protein and Chroma need to be on the same device, in this case "cuda".

ning-2022 commented 10 months ago

That sounds like the right fix, Protein and Chroma need to be on the same device, in this case "cuda".

Thanks for your reply. This program is really amazing.