使用 PyTorch 进行深度学习:60 分钟闪电教程¶
Created On: Mar 24, 2017 | Last Updated: May 31, 2023 | Last Verified: Nov 05, 2024
作者: Soumith Chintala
什么是 PyTorch?¶
PyTorch 是一个基于 Python 的科学计算包,服务于两个主要目的:
作为 NumPy 的替代品,利用 GPU 和其他加速器的性能。
一个自动微分库,用于实现神经网络。
本教程的目标:¶
从高层理解 PyTorch 的张量库和神经网络。
训练一个小型神经网络来进行图像分类
要运行以下教程,确保安装了 torch, torchvision 和 matplotlib 包。
Tensors
在本教程中,您将学习 PyTorch 张量的基础知识。
A Gentle Introduction to torch.autograd
了解 autograd。
Neural Networks
本教程展示如何在 PyTorch 中训练神经网络。
Training a Classifier
学习如何使用 CIFAR10 数据集在 PyTorch 中训练图像分类器。