#pytorch
Вопрос:
Я пытаюсь запустить PyTorch с процессором. Я создаю экземпляр устройства с помощью
device = torch.device( "cuda" if torch.cuda.is_available() else "cpu" , index = 0)
и выход print(device) есть device: cpu:0 .
Но когда я пытаюсь отправить тензоры на свое устройство .to(device) , я получаю ошибку
RuntimeError: Device index must not be negative
Есть ли другой способ вручную задать индекс устройства?
Комментарии:
1. Недостаточно информации для устранения причины проблемы. Предоставленный вами код отлично работает как без, так и без графического процессора.