PyTorch 深度学习:60 分钟速成教程¶
创建于:2017 年 3 月 24 日 | 最后更新:2023 年 5 月 31 日 | 最后验证:2024 年 11 月 05 日
作者: Soumith Chintala
什么是 PyTorch?¶
PyTorch 是一个基于 Python 的科学计算软件包,服务于两个广泛的目标
NumPy 的替代品,以利用 GPU 和其他加速器的强大功能。
一个自动微分库,可用于实现神经网络。
本教程的目标:¶
在高层次上理解 PyTorch 的张量库和神经网络。
训练一个小型神经网络来分类图像
要运行以下教程,请确保您已安装 torch、torchvision 和 matplotlib 软件包。
在本教程中,您将学习 PyTorch 张量的基础知识。
了解 autograd。
本教程演示了如何在 PyTorch 中训练神经网络。
了解如何通过使用 CIFAR10 数据集在 PyTorch 中训练图像分类器。