Stuck on an issue?

Lightrun Answers was designed to reduce the constant googling that comes with debugging 3rd party libraries. It collects links to all the places you might be looking at while hunting down a tough bug.

And, if you’re still stuck at the end, we’re happy to hop on a call to see how we can help out.

Error for cuda and cpu

See original GitHub issue

/pytorch-seq2seq/seq2seq/models/EncoderRNN.py", line 68, in forward
    embedded = self.embedding(input_var)
  File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py", line 479, in __call__
    result = self.forward(*input, **kwargs)
  File "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/sparse.py", line 113, in forward
    self.norm_type, self.scale_grad_by_freq, self.sparse)
  File "/usr/local/lib/python3.6/dist-packages/torch/nn/functional.py", line 1283, in embedding
    return torch.embedding(weight, input, padding_idx, scale_grad_by_freq, sparse)
RuntimeError: Expected object of backend CUDA but got backend CPU for argument #3 'index'

Issue Analytics

State:
Created 5 years ago
Comments:8

Top GitHub Comments

5reactions

v587sucommented, May 21, 2019

The wrong code is at line 75 of supervised_trainer.py and line 38 of evaluator.py. I change them to device = torch.device('cuda:0') if torch.cuda.is_available() else -1 and CUDA is available now.

0reactions

isekuliccommented, May 24, 2021

The wrong code is at line 75 of supervised_trainer.py and line 38 of evaluator.py. I change them to device = torch.device('cuda:0') if torch.cuda.is_available() else -1 and CUDA is available now.

Except the two files, is there any other file need to fix? After fix the two files, the code still be incorrect…

No, but you need to re-install the package after you do these corrections. Running python setup.py install again helped in my case.