深度学习与 PyTorch:60 分钟速成¶
创建于:2025 年 4 月 1 日 | 最后更新:2025 年 4 月 1 日 | 最后验证:2024 年 11 月 5 日
作者:Chintala Soumith
什么是 PyTorch?
PyTorch 是一个基于 Python 的科学计算包,具有两个广泛的应用目的:
作为 NumPy 的替代品,利用 GPU 和其他加速器的强大功能。
一个自动微分库,对于实现神经网络非常有用。
本教程的目标是:
高层次理解 PyTorch 的 Tensor 库和神经网络。
训练一个小型神经网络以进行图像分类。
运行以下教程之前,请确保已安装 torch、torchvision 和 matplotlib 包。
在本教程中,您将学习 PyTorch 张量的基础知识。
学习关于自动微分的内容。
本教程演示了如何在 PyTorch 中训练神经网络。
学习如何在 PyTorch 中使用 CIFAR10 数据集训练图像分类器。