torch.get_default_device() gets error while torch.set_default_device() doesn't

[hyperkai]

🐛 Describe the bug

torch.get_default_device() gets error while torch.set_default_device() doesn't as shown below:

import torch

torch.set_default_device(device='cuda:0')
torch.get_default_device() # Error

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

Versions

import torch

torch.__version__ # 2.2.1+cu121

[spzala]

@hyperkai hello, with the latest main (torch.__version__ 2.4.0a0+git7899034), I couldn't reproduce the error. I get the output cuda:0 when print(torch.get_default_device()) Please verify and if you are okay, we can close this issue. Thanks!

[hyperkai]

Thanks @spzala. I upgraded PyTorch from 2.2.1+cu121 to 2.3.0, then torch.get_default_device() works as shown below:

import torch

torch.set_default_device(device='cuda:0')
torch.get_default_device() # device(type='cuda', index=0)

torch.__version__ # 2.3.0

[drisspg]

I am going to close this since it looks like it was solved on newer versions of pytorch

[spzala]

Thanks for the quick reply and verifying with the newer version @hyperkai !! @drisspg sounds good to me, thank you!!