pytorch - 属性错误 : 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'

我 99% 确定这是因为我没有将 pytorch 从 0.31 升级到 0.4 但是我现在不能升级 pytorch。

我需要将 .device (0.4) 翻译成适用于 0.31 的东西。

我检查了 migration document但是它没有提供我如何在回顾时转换 torch.device 。请帮忙!

Answer 1

torch.cuda.device() 是一个上下文管理器。

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

以上从 0.2 版本开始工作。