评价此页

使用 PyTorch 进行深度学习:60 分钟闪电教程#

创建日期:2017 年 3 月 24 日 | 最后更新:2023 年 5 月 31 日 | 最后验证:2024 年 11 月 5 日

作者Soumith Chintala

什么是 PyTorch?#

PyTorch 是一个基于 Python 的科学计算包,具有两个广泛的用途:

  • 替代 NumPy,利用 GPU 和其他加速器的强大功能。

  • 一个自动微分库,可用于实现神经网络。

本教程的目标:#

  • 高层次地理解 PyTorch 的 Tensor 库和神经网络。

  • 训练一个小型神经网络来对图像进行分类

要运行下面的教程,请确保已安装 torchtorchvisionmatplotlib 包。

张量

在本教程中,您将学习 PyTorch Tensors 的基础知识。

blitz/tensor_tutorial.html
torch.autograd 简介

了解 autograd。

blitz/autograd_tutorial.html
神经网络

本教程演示了如何在 PyTorch 中训练神经网络。

blitz/neural_networks_tutorial.html
训练分类器

学习如何使用 CIFAR10 数据集在 PyTorch 中训练图像分类器。

blitz/cifar10_tutorial.html