AttributeError: module 'torch' has no attribute "device"

Question

---> 13 device = torch.device({"cuda"} if torch.cuda.is_available() else {"cpu"})
     14
     15

AttributeError: module 'torch' has no attribute 'device'

I'm 99% sure this is because I didn't upgrade pytorch from 0.31 to 0.4 however I can't upgrade pytorch for now.

I need to translate .device (0.4) to something that works in 0.31.

I check the migration document however it doesn't provide how I can convert torch.device in retrospect. Please help!

Wasi Ahmad · Accepted Answer · 2018-06-10 12:26:34Z

6

torch.cuda.device() is a context manager.

torch.cuda.set_device(0)
# On device 0
with torch.cuda.device(1):
    print("Inside device is 1")    
    # On device 1
print("Outside is still 0")
# On device 0

And the above works from 0.2 version.

answered Jun 10, 2018 at 12:26

Wasi Ahmad

38.1k34 gold badges129 silver badges169 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

szelesaron · Accepted Answer · 2020-05-19 08:38:17Z

0

In version 1.5

if torch.cuda.is_available():
    device = torch.device("cuda:0")
    print("running on the GPU")
else:
    device = torch.device("cpu")
    print("running on the CPU")

answered May 19, 2020 at 8:38

szelesaron

697 bronze badges

Collectives™ on Stack Overflow

AttributeError: module 'torch' has no attribute "device"

2 Answers 2

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related