Possible to release pypi package for windows on arm64 platform?

microsoft / DirectML

DirectML is a high-performance, hardware-accelerated DirectX 12 library for machine learning. DirectML provides GPU acceleration for common machine learning tasks across a broad range of supported hardware and drivers, including all DirectX 12-capable GPUs from vendors such as AMD, Intel, NVIDIA, and Qualcomm.

MIT License

2.24k stars 301 forks source link

Possible to release pypi package for windows on arm64 platform? #628

Closed TingyiChen closed 3 months ago

TingyiChen commented 3 months ago

I see some work for pytorch running on windows arm64 platform. However, only cpu is supported in this version. Since Qualcomm platforms are also supported by DirectML, I hope we can run pytorch natively on arm-based windows machines soon.

PR on pytorch side: https://github.com/pytorch/pytorch/pull/133088

joshjkim commented 3 months ago

@TingyiChen We're currently working on support for DirectML optimizations for GPU on ARM64 devices with torch-directml

Marking as duplicate of #366