代码之家 › 专栏 › 技术社区 › aerin

属性错误:模块“torch”没有属性“device”

pytorch

aerin · 技术社区 · 6 年前

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

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

我99%肯定这是因为我没有把火炬从0.31升级到0.4但是我现在不能升级Py火炬。

我需要将.device(0.4)转换为在0.31中工作的内容。

我检查 migration document 然而,它不能提供我如何转换火炬。请帮助!

1 回复 | 直到 6 年前

Wasi Ahmad 6 年前

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版本的。

推荐文章

Ze0ruso · 做模型。eval()在代码中的位置有什么关系?

2 年前

rkraaijveld · sklearn的Coef。线性回归为无

2 年前

Ellis Thompson · 填充一个numpym数组以满足所需的大小

2 年前

Adam · 如何在多个模型上并行运行PyTorch推理?

2 年前

quinlan · 是dgl。nn。皮托克。GraphConv聚合边缘信息?

2 年前

Ofek Nourian · 完全连通的简单自回归模型预测

2 年前

MrThiele1708 · 运行时错误:mat1和mat2形状不能相乘(5400x64和5400x64)

2 年前

Moodhi · 计算GEC的F分数

2 年前

john wang · matlab使用可调用的python函数,但需要输入

2 年前

odbhut.shei.chhele · 如何解决错误:RuntimeError:收到0项ancdata

2 年前