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

hey,

have you tried protein = Protein('1GFP', device='cuda') ?

Thanks. It works now. I thought somewhere I should specify the device, but I could not find chroma.design.

from chroma.

I did not have any problems before today I run "apt --fix-broken install" to fix some inconsistency issue.

Here is the apt log file (attached).
history.log

from chroma.

The problem happens when chroma.design() is used. I found cuda was not available before the nvidia driver was updated, therefore all tensors used 'cpu'. Now cuda is available, the RuntimeError persists; there is no problem with other examples except for the redesign.

from chroma.

hey,

have you tried protein = Protein('1GFP', device='cuda') ?

from chroma.