#pytorch #amd
#пыторч #драм
Вопрос:
Согласно официальным документам, теперь PyTorch поддерживает графические процессоры AMD. ROCm 4.2 может быть установлен через pip. Но я не могу найти ни в Google, ни в официальных документах, как заставить мое обучение DL использовать графический процессор. Что эквивалентно AMD следующей команде?
torch.device('cuda' if torch.cuda.is_available() else 'cpu')