Construcciones Nativas de PyTorch para Arm Ya Disponibles en Windows

0
4
PyTorch Arm native builds now available for Windows

La comunidad de desarrollo y la investigación en inteligencia artificial se encuentran de enhorabuena con el reciente lanzamiento de versiones nativas de PyTorch para Windows en arquitectura Arm. Hasta ahora, los usuarios de Windows Arm64 se veían obligados a compilar PyTorch de forma local para poder utilizarlo, pero con la versión 2.7 de PyTorch, esta complejidad se ha eliminado. Ahora es posible acceder a estas versiones nativas optimizadas para Python 3.12, lo que permite a los desarrolladores aprovechar completamente el rendimiento de la arquitectura Arm64 en dispositivos Windows, como los PCs Copilot+.

Con las nuevas versiones de PyTorch, los usuarios pueden desarrollar, entrenar y probar modelos de aprendizaje automático de pequeña escala directamente en sus PCs con arquitectura Arm. Este avance resulta crucial en áreas como la clasificación de imágenes, el procesamiento de lenguaje natural y la inteligencia artificial generativa, incluyendo aplicaciones como Stable Diffusion.

Para los desarrolladores interesados, se recomienda instalar herramientas adicionales como MSVC y Rust para abordar posibles dependencias faltantes. También hay que asegurarse de seleccionar las herramientas de compilación C++ ARM64 de Visual Studio 2022 durante la instalación.

Para instalar la versión estable de PyTorch (2.7.0), se puede usar el siguiente comando en la terminal: pip install --extra-index-url https://download.pytorch.org/whl torch. Aquellos que deseen acceder a la versión preliminar (Nightly) deben usar otro comando específico, facilitando una transición fluida a las nuevas características en desarrollo.

Además, se presenta un ejemplo práctico del uso de las nuevas versiones nativas de PyTorch, ejemplificado mediante el modelo de difusión estable "stabilityai/sd-turbo". El modelo nos permite generar imágenes a partir de descripciones textuales y ofrece controles sobre los pasos de inferencia y valores semilla para garantizar la reproducibilidad.

Sin embargo, es importante señalar que algunos paquetes complementarios que los desarrolladores pueden desear usar con PyTorch aún no tienen soporte nativo para Windows en arquitectura Arm. La instalación de dependencias que carezcan de soporte puede realizarse mediante la compilación de código fuente, gracias a herramientas como MSVC y Rust.

En resumen, el lanzamiento de estas versiones nativas de PyTorch para Windows abre un nuevo horizonte para los desarrolladores, permitiéndoles explorar y aprovechar al máximo la arquitectura Arm en sus proyectos de inteligencia artificial. Las invitaciones están abiertas para que los desarrolladores descarguen y prueben estas versiones para impulsar sus innovaciones.

vía: Microsoft Windows blog