viable/strict/1787131695: Support Windows ARM64 CUDA paths in cpp_extension (#191656)
PyTorch 的 cpp_extension 模块新增对 Windows ARM64 CUDA 路径的支持，在构建时识别 win-arm64 VS 目标，并使用 lib/arm64 作为 CUDA 和 cuDNN 库路径，同时保持现有 x64 行为不变。该改动通过 PR #191656 合并，由 Nikita Shulga 共同编写。
Development
- First Reportviable/strict/1787131695: Support Windows ARM64 CUDA paths in cpp_extension (#191656)PyTorch Core
- Current AssessmentPyTorch 对 Windows ARM64 的支持反映了 AI 基础设施向 ARM 架构扩展的趋势，可能为 ARM 设备上的 AI 开发提供更多可能性。这可能会影响开发者对开发环境的选择，并推动相关工具链的适配。Agent Pulse · analysis
PyTorch 在 2026 年 8 月 19 日发布的版本中，通过 PR #191656 为 cpp_extension 增加了对 Windows ARM64 CUDA 路径的支持。该改动使 cpp_extension 能够识别 win-arm64 Visual Studio 目标，并在构建扩展时使用 lib/arm64 目录下的 CUDA 和 cuDNN 库，避免错误地查找 x64 目录。此改动不影响现有的 x64 构建行为。
该改动表明 PyTorch 正在适配 Windows ARM64 平台，这需要处理 CUDA 库路径的差异。对于开发者而言，这意味着在 Windows ARM64 上构建自定义扩展时，cpp_extension 将自动选择正确的库路径，减少了手动配置的需要。未来可关注 PyTorch 对 ARM64 的进一步支持，如是否支持 CUDA 的更多功能。
PyTorch 对 Windows ARM64 的支持反映了 AI 基础设施向 ARM 架构扩展的趋势，可能为 ARM 设备上的 AI 开发提供更多可能性。这可能会影响开发者对开发环境的选择，并推动相关工具链的适配。
该改动降低了在 Windows ARM64 上开发 PyTorch 扩展的门槛，有助于吸引更多开发者使用 ARM 设备进行 AI 开发，可能扩大 PyTorch 在 ARM 生态中的影响力，并促进相关硬件和软件的销售。
未来，PyTorch 可能会继续完善对 Windows ARM64 的支持，包括优化性能、支持更多 CUDA 功能，并可能扩展到其他库。开发者可以关注 PyTorch 的发布说明，以了解 ARM64 支持的进展。