PyTorch on Windows on Arm64

Owned by Everton Constantino
Last updated: Aug 30, 2024 by Alina Taus
2 min read

Project

Focus is to port PyTorch on Windows on Arm64, with the first target of CPU only. Acceleration will follow after. The focus is on local experimentation and small scale training. PyTorch on Windows on Arm64 should offer a comparable experience, regardless of the underlying CPU architecture. Users typically rely on a set of libraries such as TorchAudio, TorchVision etc. These libraries are key to enabling experimentation and professional development. 

Status

[August 2024]

  • MSFT is continuing support for PyTorch on Windows, with collaboration with Meta

  • PyTorch on Windows on Arm64 with APL, CPU only is now publicly opened: Enable Windows Arm64. In review with Meta and Arm

  • Internal and external CI/CD is in progress

  • TorchAudio and TorchVision enablement are in progress - targeting build for now