小编Jia*_*ang的帖子

我是深度学习领域的新手。现在我正在复制论文的代码。由于他们使用多个 GPU，因此torch.nn.DataParallel(model, device_ids= args.gpus).cuda()代码中有一个命令。但是我只有一个 GPU，我应该如何更改此代码以匹配我的 GPU？

谢谢！