search

microsoft / evodiff

Generation of protein sequences and evolutionary alignments via discrete diffusion models
MIT License
483 stars 67 forks source link

tag device='cuda' not working #9

Closed arjunsbalaji closed 1 month ago

arjunsbalaji commented 1 year ago

Hi,

When I specify device=cuda I get an error saying cpu and cuda devices are in use?

I can see that in generate.py .to(device) is being called in inputs and model so not sure why this is happening.

sarahalamdari commented 11 months ago

Can you specify which script/function you are running? And I can help solve this

colbyford commented 10 months ago

@arjunsbalaji - When running from Python, you can run the following command to set the default Torch device:

import torch
torch.set_default_device('cuda:0')

This fixed the cpu/cuda error for me...

yangkky commented 10 months ago

@arjunsbalaji do you have an update or more details? Otherwise, I would like to close this.

yangkky commented 10 months ago

In general, I believe you have to set device='cuda:0', not just cuda.