小编ML *_* Xu的帖子

我正在学习pytorch并遵循此教程：https ://pytorch.org/tutorials/beginner/transfer_learning_tutorial.html

我使用此命令来使用GPU。

device = torch.device("**cuda:0**" if torch.cuda.is_available() else "cpu")

但是，我想在jupyter中使用两个GPU ，如下所示：

device = torch.device("**cuda:0,1**" if torch.cuda.is_available() else "cpu")

当然，这是错误的。那么，我该怎么做？