使用 PyTorch 进行深度学习:60 分钟快速入门¶
作者: Soumith Chintala
什么是 PyTorch?¶
PyTorch 是一个基于 Python 的科学计算包,服务于两个广泛的目的
替代 NumPy 以利用 GPU 和其他加速器的强大功能。
一个自动微分库,可用于实现神经网络。
本教程的目标:¶
从高级别了解 PyTorch 的张量库和神经网络。
训练一个小型的分类图像的神经网络
要运行下面的教程,请确保已安装 torch、torchvision 和 matplotlib 包。
在本教程中,您将学习 PyTorch 张量的基础知识。
了解自动微分。
本教程演示了如何在 PyTorch 中训练神经网络。
通过使用 CIFAR10 数据集,学习如何在 PyTorch 中训练图像分类器。