AMD anuncia suporte RDNA3 para PyTorch e ROCm
A AMD anunciou que três de suas placas gráficas de desktop RDNA3, a Radeon RX 7900 XT, 7900 XTX e a Radeon Pro 7900, agora ofereceram suporte ao desenvolvimento de aprendizado de máquina por meio de PyTorch e seu software ROCm. PyTorch foi originalmente feito pela Meta AI, mas atualmente faz parte da Linux Foundation. Portanto, é uma estrutura de código aberto para a construção de modelos de aprendizado profundo, um tipo de aprendizado de máquina usado em aplicativos de reconhecimento de imagem e processamento de linguagem. (Crédito da imagem: AMD)
As respectivas placas gráficas RDNA 3 Radeon aproveitam o AMD ROCm 5.7, que é exclusivo para Ubuntu 22.04. A AMD fornece instruções detalhadas em um guia prático. (Crédito da imagem: AMD)
Requisitos de hardware de memória AMD ROCm
A AMD tem uma lista de pré-requisitos para suporte ROCm, um dos quais exige que o usuário desative o iGPU em duas placas-mãe AMD x670 específicas – X670 Aorus Elite AX da Gigabyte e Asus Prime X670-P WIFI. O designer do chip GPU tem duas recomendações de requisitos de memória – você precisará de um sistema com 64 GB de RAM e GPUs para ter GPU VRAM de 24 GB conforme recomendado pela AMD, enquanto o requisito mínimo de memória é 16B de memória de sistema e 8GB de memória de vídeo. A GPU RX 7900 XT possui 20 GB de memória GDDR6, o que se enquadra nessas recomendações, enquanto a Radeon RX 7900 XTX atende aos requisitos de VRAM recomendados. A placa gráfica da estação de trabalho Pro W7900 excede esse requisito, pois possui uma pilha de chips GDDR6 de 48 GB. AMD Radeon 7900 XT utiliza 168 aceleradores AI baseados no núcleo Navi 31 de 5 nm fabricado pela TSMC. Há alguns meses, a AMD listou algumas placas gráficas não anunciadas em sua atualização ROCm 5.6 – a Radeon RX 7950 XTX, 7950 XT, 7800 XT, 7700 XT, 7600 XT e 7500 XT para desktops e a Radeon RX 7600M XT, 7600M, 7700S e 7600S para notebooks. Portanto, é provável que, se e quando essas GPUs forem anunciadas e lançadas, elas sejam habilitadas com suporte no primeiro dia.
A curiosa história do RX 7900 GRE
A Radeon RX 7900 GRE possui 160 aceleradores e 16 GB de memória de vídeo. Por outro lado, a Radeon RX 7900 GRE é uma GPU exclusiva da China (Golden Rabbit Edition já que 2023 é o ano do coelho) e só pode ser adquirida por meio de integradores de sistemas no mercado doméstico. Os EUA proibiram certos aceleradores de IA na China. Não se sabe se o driver ROCm funcionará nesta placa gráfica ‘Golden Rabbit’. Teremos que esperar para ver.
Do outro lado da lagoa
A Nvidia também suporta PyTorch, e assim como a AMD, requer o Docker Engine. Também requer drivers de GPU da Nvidia e seu Container Toolkit. PyTorch aproveita os núcleos Nvidia Tensor e os aceleradores de IA da AMD. A Intel também suporta PyTorch e possui extensa documentação para habilitar isso em sistemas compatíveis. A nova extensão de suporte para todos os três fabricantes de chips gráficos ajudará muitos desenvolvedores a acelerar seu aprendizado de máquina e desenvolvimento de IA. Pode-se estar curioso para saber se a ativação de GPUs de última geração resultará em um preço mais alto para esses gráficos. No entanto, ainda não há sinais de que isso aconteça, mas não seria sem precedentes: vimos tendências semelhantes quando placas gráficas para desktop são usadas para outros fins, como a criptomineração de GPU.