Descrição
Fornece uma build PyTorch com suporte CUDA de GPU e otimizações AVX2 de CPU. Ajuda sistemas compatíveis a acelerar treinamento de modelos, inferência e computação com tensores em GPUs Nvidia e CPUs modernas.
Este runtime é sensível ao hardware. Confira compatibilidade de driver, suporte de instruções da CPU, memória de GPU e limites de recursos antes de usar em cargas grandes.