Running on CPU

[rkruegs123]

Hi, when running the test code on a CPU I got the following error: AttributeError: module 'torch._C' has no attribute '_cuda_setDevice'

I am guessing that the code assumes that one has a GPU, but I don't see much reason for this if I am just using a pre-trained model for inference.

[sperfu]

Hi,

I guess it's because you only run the code on CPU. In that case, you can alter the code:

UFold/ufold_predict.py

Line 297 in 2b5fab2

device = torch.device("cuda:1" if torch.cuda.is_available() else "cpu")

to:

  device = torch.device("cpu")

to avoid this issue.

Please have a try.

Thanks

[IseultLeahy]

Hey,
I've been getting the same error message and tried your solution but unfortunately it didn't change anything...

UFold/ufold_predict.py

Line 297 in 2b5fab2

device = torch.device("cuda:1" if torch.cuda.is_available() else "cpu")

to:

  device = torch.device("cpu")