AMD ha presentado un controlador ROCm 5.7.1 actualizado para Ubuntu Linux que brinda compatibilidad con PyTorch 2.0.1 para tarjetas gráficas Radeon Pro W7900 y Radeon RX 7900 XTX basadas en RDNA 3. La medida permite a los desarrolladores utilizar las GPU cliente insignia de AMD con 24 GB y 48 GB de memoria para aplicaciones de inteligencia artificial que utilizan bibliotecas PyTorch.
«Estamos entusiasmados de ofrecer a la comunidad de IA un nuevo soporte para el desarrollo de aprendizaje automático utilizando PyTorch integrado en las GPU AMD Radeon RX 7900 XTX y Radeon Pro W7900 y la plataforma de software abierto ROCm. Esta es nuestra primera implementación basada en la arquitectura RDNA 3, y estamos Esperamos asociarnos con la comunidad», afirma Dan Wood, vicepresidente de gestión de productos Radeon de AMD.
La actualización del controlador está diseñada específicamente para funcionar con PyTorch 2.0.1, una popular biblioteca de aprendizaje profundo, que ofrece capacidades de las GPU basadas en RDNA3 de AMD. El controlador ROCm 5.7.1 solo admite tarjetas gráficas emblemáticas: Radeon RX 7900 XTX de 24 GB y Radeon Pro W7900 de 48 GB. Esto es lógico ya que las aplicaciones de IA, como los modelos de lenguaje grande (LLM), se benefician de la memoria integrada de alta capacidad y AMD quiere vender tantas placas de alta gama como sea posible.
AMD señala que su procesador gráfico Navi 31 tiene 192 aceleradores de IA integrados, frente a 96 en el caso de Navi 21, lo que lo convierte en una GPU viable para aplicaciones de inteligencia artificial. Mientras tanto, la Radeon RX 7900 XTX de 24 GB de AMD es una tarjeta gráfica para el consumidor con un MSRP de $999, que es una fracción del costo de la IA profesional y la GPU HPC.
AMD facilita una convergencia matizada de las funcionalidades de GPU para juegos y profesionales al extender el soporte sofisticado de ML tradicionalmente reservado para dominios profesionales a productos de consumo. AMD espera fomentar un ecosistema enriquecido donde prosperen las plataformas de código abierto, allanando el camino para una mayor rentabilidad y una mayor accesibilidad.