探索为Ultralytics YOLO26提供动力的核心库PyTorch。了解其动态图、GPU加速以及如何构建高效的深度学习模型。
PyTorch 是一个主要由 Meta AI 开发的开源机器学习库,已成为深度学习领域研究人员和开发者的基石。它以其灵活性和易用性而闻名,允许用户使用动态计算图构建和训练复杂的 神经网络。这一特性,常被称为“即时执行”,使代码能够立即被评估,与依赖静态图定义的框架相比,显著提高了 调试 和原型开发的直观性。它与 python 编程语言无缝集成,感觉就像是标准科学计算工具的自然延伸。
该框架的核心是tensor,它们是多维数组,类似于NumPy documentation中描述的数组。然而,与标准数组不同,PyTorch tensor旨在利用NVIDIA CUDA提供的GPU加速。这种硬件加速对于高效训练现代人工智能 (AI)模型所需的大规模并行处理至关重要。
该库支持 计算机视觉 (CV) 和自然语言处理的庞大工具生态系统。通过提供丰富的预构建层、优化器和损失函数,它简化了创建用于 图像分类 和序列建模等任务的算法的过程。
该框架的多功能性使其在不同行业中得到采用,用于高影响力的AI解决方案:
为了更好地理解其作用,有必要将 PyTorch 与 AI 技术栈中的其他常用工具区分开来:
整个 Ultralytics 模型家族,包括尖端的YOLO26和广泛使用的YOLO11,都是原生构建在 PyTorch 之上。这一基础确保用户能够从该框架的速度、稳定性和广泛的社区支持中获益。无论是对自定义训练数据进行迁移学习,还是将模型导出到边缘设备,其底层架构都依赖于 PyTorch 的 tensor 和梯度。
即将推出的Ultralytics Platform进一步简化了这一体验,提供了一个统一的界面,无需编写大量样板代码即可管理数据集获取、训练和部署。
以下示例演示了如何验证GPU可用性并使用YOLO模型运行推理,展示了该框架如何在底层处理硬件加速:
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)

开启您的机器学习未来之旅