PyTorch 入门指南
学习 PyTorch
图像和视频
音频
后端
强化学习
在生产环境中部署 PyTorch 模型
Profiling PyTorch
代码变换与FX
前端API
扩展 PyTorch
模型优化
并行和分布式训练
边缘端的 ExecuTorch
推荐系统
多模态

使用 PyTorch 进行深度学习:60 分钟快速入门

作者: Soumith Chintala

什么是 PyTorch?

PyTorch 是一个基于 Python 的科学计算库,主要服务于两大用途:

  • 一个替代 NumPy 的工具,能够利用 GPU 和其他加速器的强大性能。

  • 一个自动微分库,对于实现神经网络非常有用。

本教程的目标:

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

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

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

张量

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

代码

blitz/tensor_tutorial.html

torch.autograd 简明介绍

了解 autograd。

代码

blitz/autograd_tutorial.html

神经网络

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

代码

blitz/neural_networks_tutorial.html

训练分类器

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

代码

blitz/cifar10_tutorial.html

本页目录