Как использовать NeuralForecast и PyTorch Lightning на графическом процессоре Intel (XPU/torch.xpu)?Python

Программы на Python
Anonymous
Как использовать NeuralForecast и PyTorch Lightning на графическом процессоре Intel (XPU/torch.xpu)?

Сообщение Anonymous »

PyTorch поддерживает графический процессор Intel через torch.xpu, но PyTorch Lightning в настоящее время не имеет встроенной поддержки ускорителя XPU.
Поскольку NeuralForecast использует Lightning под капотом, это также блокирует NeuralForecast на собственном XPU.
Мне нужно минимальное рабочее решение для:
  • PyTorch Lightning Тренажер
  • Nixtla NeuralForecast
  • native torch.xpu
  • один графический процессор Intel / одно устройство XPU
  • без IPEX
Какой самый простой обходной путь работает сегодня?

Вернуться в «Python»