微軟推出 Windows 的 Arm 原生 PyTorch 版本
在向 GitHub 推出 Windows on Arm 運行器支援僅幾週後，微軟就向針對運行 Windows 的 Arm 驅動裝置的開發人員發布了一項具有里程碑意義的聲明。PyTorch是一個廣泛使用的開源機器學習框架，其原生版本現已正式可用於Arm 上的 Windows。這項進步簡化了旨在在 Arm 架構上原生建置和測試應用程式的開發人員的流程。
PyTorch 對機器學習的意義
對於那些不熟悉的人來說，PyTorch 是專注於建立和訓練深度神經網路的研究人員和開發人員的重要工具。從歷史上看，在配備 Arm 晶片的 Windows 裝置上運行 PyTorch 是一項挑戰，因為用戶需要從原始碼編譯整個框架——這個過程非常耗時，而且不太適合程式新手。
PyTorch 2.7 中的新功能
隨著PyTorch 2.7的發布，適用於 Arm 上的 Windows 的原生版本現在可用於Python 3.12。開發人員可以使用標準套件管理器（如 pip ）輕鬆安裝 PyTorch ，從而大幅簡化設定流程。
據微軟表示：
這釋放了在 Windows 設備（如 Copilot+ PC）上充分利用 Arm64 架構的全部性能進行機器學習實驗的潛力，為開發人員和研究人員提供了一個強大的平台來創新和改進他們的模型。
機器學習開發的好處
預計這項最新產品將直接在基於 Arm 的 Windows 機器上增強機器學習模型的本地開發、訓練和測試。值得注意的是，微軟強調了影像分類、自然語言處理和生成式人工智慧等領域的潛在應用，例如 Stable Diffusion 等工具。
在 Windows 上開始使用原生 PyTorch
為了開始使用適用於 Arm 上的 Windows 的原生 PyTorch 二進位文件，開發人員必須安裝一些必要的先決條件。這包括從Visual Studio Build Tools或完整的Visual Studio 安裝中獲得的元件。
- 確保選擇使用 C++ 進行桌面開發工作負載。
- 確保在安裝過程中包含最新的VS 2022 C++ ARM64/ARM64EC建置工具。
此外，還需要在系統上安裝Rust並安裝 Arm64 版本的 Python 3.12。一旦滿足這些先決條件，一個簡單的命令將使您能夠透過 pip 安裝穩定版本的 PyTorch，並針對正確的下載索引：
pip install --extra-index-url https://download.pytorch.org/whl torch
探索夜間構建
對於那些渴望嘗試最新功能並且對可能不穩定的版本感到滿意的人，可以使用以下命令安裝 Nightly 或 Preview 版本：
pip install --pre torch --index-url https://download.pytorch.org/whl/nightly/cpu
其他原生建置和最佳實踐
除了 Python 套件之外，LibTorch 的原生建置也已推出，LibTorch 是 PyTorch 的 C++ 前端，經常用於部署環境。如需 LibTorch 入門的綜合指南，開發人員可以造訪 PyTorch 網站。作為 Python 開發的標準實踐，微軟建議創建虛擬環境 (venv)來無縫管理專案依賴關係並避免潛在的衝突。
應用範例和採用
此外，微軟還展示了在 Arm 上的 Windows 上使用原生 PyTorch 二進位檔案進行穩定擴散的範例，為開發人員如何在其應用程式中利用生成式 AI 提供了見解。相關程式碼可以在這個GitHub 倉庫中找到。
解決依賴性挑戰
值得注意的是，雖然 PyTorch 和 LibTorch 現在支援 Arm 上 Windows 的本機二進位文件，但並非所有依賴項都支援。一些額外的 Python 包，特別是那些用 C、C++ 或 Rust 等語言編寫的效能敏感元件的包，可能還沒有提供預編譯的本機 Arm64。 PyPI 上的 whl 檔。因此，簡單的 pip 安裝可能不會為每個正在使用的程式庫產生本機版本。
但是，pip 可以直接從通常提供的原始程式碼分發中安裝依賴項。 tar.gz 文件。如果您的系統上有適當的建置工具（透過 Arm64 工具鏈和 Rust 強化前面提到的 MSVC），pip 可以將這些套件編譯成與 Windows 相容的。 whl 檔案在本地。
微軟強調，此方法允許安裝流行軟體包的特定版本，例如 NumPy 2.2.3 和 safetensors 0.5.3，並展示了所需的命令：
pip install numpy==2.2.3 # and pip install safetensors==0.5.3
這些命令是如何有效地從原始碼編譯套件的範例。
閱讀更多
如需進一步見解和其他範例，請參閱Microsoft Windows 部落格上的完整公告。
發佈留言 ▼