Windows now supports PyTorch 2.2 with DirectML, enabling developers to easily run AI models across all GPUs. The update simplifies setup and scales AI across Windows, promising a seamless experience with the latest Generative AI models.-

Unlocking AI’s Next Frontier on Windows

Windows developers now have a powerful tool at their disposal for AI development, thanks to the integration of PyTorch 2.2 with DirectML. This collaboration marks a significant milestone in making advanced AI models more accessible across the Windows ecosystem.

What’s New: PyTorch 2.2 Meets DirectML

With the latest update, PyTorch leverages DirectML 1.13 for enhanced acceleration, simplifying the AI development process on Windows. This update promises a one-package install, streamlining the setup process for developers.

“We’ve updated Torch-DirectML to use DirectML 1.13 for acceleration and support PyTorch 2.2.” – Clarke Rahrig and Sheil Kumar

Major Updates: A Seamless AI Development Experience

The integration of PyTorch with DirectML is designed to provide a seamless experience for developers working with the latest Generative AI models. This update ensures optimal performance and scalability across various Windows devices.

Why It’s Important: Broad GPU Support

This update extends the capabilities of Windows developers to run inference on a wide range of GPUs, making AI development more versatile and inclusive.

Easy Installation and Usage

Getting started with PyTorch and DirectML on Windows is as easy as running a simple pip install command, followed by a quick setup to run AI models locally.

“…using the latest Torch-DirectML package with your Windows GPU is as simple as running: pip install torch-directml” – Windows Developer Blog

Partnerships That Matter

AMD, Intel, and NVIDIA have expressed their support for this initiative, highlighting the collaborative effort to enhance AI development on Windows. Each partner brings unique contributions, from AMD’s investment in DirectML to NVIDIA’s acceleration through RTX GPUs.

Generative AI Models at Your Fingertips

The update shines with its support for cutting-edge Generative AI models, including Llama 2, Llama 3, and Phi 3 Mini, promising an expansion in model coverage in the near future.

Get Started Today

Developers are encouraged to try out the new PyTorch with DirectML update. With easy installation and a range of supported models, it’s an opportune time to explore AI development on Windows.

In conclusion, the integration of PyTorch 2.2 with DirectML is a game-changer for AI development on Windows. It simplifies the process, broadens GPU support, and fosters collaboration with key industry players. As AI continues to evolve, Windows developers are well-equipped to lead the charge.

DirectML 1.13 accelerates PyTorch 2.2, enhancing AI model inference on Windows.

• One-package install of Torch-DirectML simplifies AI development on Windows.
• Hardware partners like AMD, Intel, and NVIDIA support PyTorch with DirectML, enhancing local LLM running.
• Generative AI model support includes Llama 2, Llama 3, Mistral, Phi 2, and Phi 3 Mini.
• PyTorch with DirectML optimizes for the latest GPUs, ensuring high performance on Windows.

From the Windows Blog

---