У меня два графических процессора, и оба они являются устройствами CUDA, но я не знаю, как использовать их оба для обучения моей модели resnext: ссылка для репозитория github, который я взял в качестве базового кода для моего проекта. Единственное, что я изменил, это:
resnet152_model = resnet.resnext50_32x4d(pretrained=True)
model = resnet152_model
model = nn.DataParallel(model).to(device)
где мое устройство:
device = torch.device('cuda')
если я напишу cuda, оно должно использовать все доступные графические процессоры, но это не так. . Он использует мой первый графический процессор, и он будет использовать только мой второй графический процессор, если я напишу:
torch.cuda.set_device(1)
В чем моя ошибка и как заставить мой код использовать несколько графических процессоров