how to run translate.py without GPU?

Question

abhishekgoldypathak opened this issue 4 years ago · 1 comments

Is it possible to run using only CPU? I have tried to run it by changing all instances of

device='cuda:0'

to device='cpu'

or setting CUDA_VISIBLE_DEVICES="" etc

but nevertheless not able to git rid of all the errors regardless of what I try. What's the easy way to run it on a CPU only machine?

Answer 1 · 2021-04-06T14:02:22.000Z

Did you also change all the instances where the .cuda() is called on a tensor? For instance in the to_cuda(*args) method.