Pytorch：如何将模型动物园预先训练的模型映射到新的GPU

我正在尝试加载一个预先训练过的模型

model_url={'resnet18'：'https://download.pytorch.org/models/resnet18-5c106cde.pth“，“resnet34”：”https://download.pytorch.org/models/resnet34-333f7ec4.pth'}

当我使用以下代码时，它总是将模型加载到cuda:0。如果我想把它加载到cuda:3怎么办

model=ResNet（基本块[3,4,6,3]）
设备=3
model.load_state_dict（model_zoo.load_url（model_url['resnet34']），在，
map_location=lambda存储，loc:storage.cuda（设备）））

这将为您完成以下任务：

device=torch.device（'cuda'））
model=ResNet（基本块[3,4,6,3]）
带火炬.cuda.装置（3）：
model.load_state_dict（model_zoo.load_url（model_url['resnet34']），在，
map_location=lambda存储，loc:storage.cuda（设备）））

我认为这适用于0.4.0及更高版本，您可以在0.4.0中查看更多示例。迁移指南：