Как использовать несколько графических процессоров в Pytorch? ⇐ Python

[Anonymous]

Я использую эту команду для использования графического процессора.device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")

Но я хочу использовать два графических процессора в jupyter , например:
device = torch.device("cuda:0,1" if torch.cuda.is_available() else "cpu")


Подробнее здесь: https://stackoverflow.com/questions/542 ... in-pytorch