Pytorch深度学习框架入门:60分钟 blitz¶
创建日期: 2017年3月24日 | 最后更新日期: 2023年5月31日 | 最后验证日期: 2024年11月5日
作者: Soumith Chintala
什么是PyTorch?¶
PyTorch 是一个基于 Python 的科学计算包,具有两大主要用途:
用于在GPU和其他加速器上利用其强大计算能力的NumPy替代品。
一个用于实现神经网络的自动微分库。
本教程的目标:¶
了解PyTorch的张量库和神经网络的基本概念。
训练一个小型神经网络来分类图像
要运行以下教程,请确保已安装 torch, torchvision, 和 matplotlib 包。
在本教程中,您将学习PyTorch张量的基本知识。
了解自动微分。
这个教程演示了如何在 PyTorch 中训练神经网络。
了解如何使用 PyTorch 和 CIFAR10 数据集训练图像分类器。