Descrição
Fornece recursos do torchvision com suporte a GPU para fluxos PyTorch de visão computacional. Ajuda sistemas compatíveis a acelerar treino de modelos de visão, inferência e operações de imagem quando recursos CUDA estão disponíveis.
Desenvolvedores e pesquisadores usam isto como biblioteca. O comportamento em GPU depende de drivers, versões de CUDA e memória do hardware, então teste compatibilidade e monitore uso de recursos na máquina de destino.