Descrição
Fornece uma build PyTorch com suporte ROCm de GPU e otimizações AVX2 de CPU. Ajuda sistemas com GPU AMD compatível a acelerar computação com tensores, treinamento e inferência a partir de Python.
Suporte ROCm depende do modelo da GPU, kernel, pilha de drivers e memória disponível. Confirme compatibilidade antes de planejar grandes cargas de aprendizado de máquina.