PyTorch 找不到名字？？ NameError：未定义名称“设备” 对不起 - 我是一个完整的初学者！

编程问答 2022-05-18

问题描述

对不起 - 我是一个完整的初学者！

我正在尝试使用 https://kaiyangzhou.github.io/deep-person-reid/index.html# 中的 Torchreid 库构建一个“迷你系统”

在他们的版本中，他们使用 CUDA，但我的 Mac 与 CUDA 不兼容，而且它没有 启用了 CUDA 的 GPU，因此我安装了仅 cpu 版本的 PyTorch - 因此我将 model = model.cuda() 更改为 model = model.to(device) 并添加到 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu') 中，如下所示。我认为这会奏效，但我一直收到 NameError: name 'device' is not defined，但我不知道该怎么办。 请帮忙！

（我还尝试将 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu') 放在顶部而不是底部以查看是否有任何区别，但我又遇到了另一个错误 - NameError: name 'torch' is not defined）

model = torchreid.models.build_model(
    name='resnet50',num_classes=datamanager.num_train_pids,loss='softmax',pretrained=True
)

model = model.to(device)

optimizer = torchreid.optim.build_optimizer(
    model,optim='adam',lr=0.0003
)

scheduler = torchreid.optim.build_lr_scheduler(
    optimizer,lr_scheduler='single_step',stepsize=20
)

device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')

解决方法

在使用前定义 device 变量：

import torch
...
model = torchreid.models.build_model(
    name='resnet50',num_classes=datamanager.num_train_pids,loss='softmax',pretrained=True
)

# Just right before the actual usage
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
model = model.to(device)
...
,

导入火炬模块。

放

import torch

在代码的顶部。

（请记住始终导入您正在使用的库）

devicedevicedevicegpugpupythonpytorch