pytorch version?

[bruce1408]

when i run the code, there is error :RuntimeError: CUDA error: invalid device ordinal

[thschaaf]

I ran into the same issue. My solution was to changing DEVICES = [2] to DEVICES = [0] in config.py.

If you are using a newer PyTorch version (e.g. 1.1) you also need to fix a few lines related to getting at the value of a tensor. The error message from PyTorch are very helpful here. For example in data_loader.py you need to replace idx = idx.data[0] with idx = idx.data.item() and in 'Train.ipynb' this_loss = loss.data[0] with this_loss = loss.data.item() .