Descubra PyTorch, a estrutura de aprendizado de máquina flexível e Python que alimenta inovações de IA como o Ultralytics YOLO. Construa de forma mais inteligente e rápida hoje mesmo!
PyTorch is an open-source machine learning library primarily developed by Meta AI that has become a cornerstone for researchers and developers in the field of deep learning. Known for its flexibility and ease of use, it allows users to build and train complex neural networks using a dynamic computational graph. This feature, often referred to as "eager execution," enables code to be evaluated immediately, making debugging and prototyping significantly more intuitive compared to frameworks that rely on static graph definitions. It integrates seamlessly with the Python programming language, feeling like a natural extension of standard scientific computing tools.
At the heart of this framework are tensors, which are multi-dimensional arrays similar to those found in the NumPy documentation. However, unlike standard arrays, PyTorch tensors are designed to leverage GPU acceleration provided by NVIDIA CUDA. This hardware acceleration is critical for the massive parallel processing required to train modern artificial intelligence (AI) models efficiently.
The library supports a vast ecosystem of tools for computer vision (CV) and natural language processing. By providing a rich set of pre-built layers, optimizers, and loss functions, it simplifies the process of creating algorithms for tasks like image classification and sequence modeling.
A versatilidade desta estrutura levou à sua adoção em diversos setores para soluções de IA de alto impacto:
Para entender melhor o seu papel, é útil distinguir PyTorch outras ferramentas comuns na pilha de IA:
Toda a família Ultralytics , incluindo o inovador YOLO26 e o amplamente utilizado YOLO11, é construída nativamente no PyTorch. Essa base garante que os utilizadores se beneficiem da velocidade, estabilidade e amplo suporte da comunidade do framework. Seja realizando aprendizado de transferência em dados de treinamento personalizados ou exportando modelos para dispositivos de ponta, a arquitetura subjacente depende de PyTorch e gradientes PyTorch .
A próxima Ultralytics simplifica ainda mais essa experiência, fornecendo uma interface unificada para gerenciar a obtenção, o treinamento e a implementação de conjuntos de dados sem a necessidade de escrever um código boilerplate extenso.
O exemplo a seguir demonstra como verificar GPU e executar a inferência usando um YOLO , mostrando como a estrutura lida com a aceleração de hardware nos bastidores:
import torch
from ultralytics import YOLO
# Check if CUDA (GPU) is available for PyTorch acceleration
device = "cuda" if torch.cuda.is_available() else "cpu"
print(f"PyTorch is using device: {device}")
# Load a YOLO26n model (built on PyTorch)
model = YOLO("yolo26n.pt")
# Perform object detection on an image
# PyTorch handles tensor operations and moves data to the GPU automatically
results = model("https://ultralytics.com/images/bus.jpg", device=device)