Seguindo a PyTorch Foundation falando sobre o PyTorch 2.0 desde o final do ano passado, ontem marcou o lançamento oficial do lançamento do PyTorch 2.0. O PyTorch 2.0 possui otimizações significativas para "sobrecarregá-lo" com melhor desempenho para os modos de operação da CPU e da GPU.
O PyTorch 2.0 usa torch/compie como a sua API principal,o TorchInductor com as GPUs NVIDIA e AMD agora conta com o OpenAI Triton para o compilador de aprendizagem profunda, um back-end Metal Performance Shaders fornece um PyTorch acelerado por GPU no macOS, com um desempenho de inferência mais rápido em CPUs AWS Graviton , e uma variedade de outros novos recursos e tecnologias de protótipo. Para uma operação mais rápida da CPU, também há várias otimizações "críticas" para a inferência e treinamento do GNN, bem como um desempenho mais rápido da CPU, fazendo uso do Intel oneDNN Graph.
Os Downloads e mais detalhes sobre o PyTorch 2.0 podem ser vistos via PyTorch.org .
Até a próxima !!
Nenhum comentário:
Postar um comentário