使用PyTorch进行深度学习:60分钟速成¶
创建于:2017年3月24日 | 最后更新:2023年5月31日 | 最后验证:2024年11月5日
作者: Soumith Chintala
什么是PyTorch?¶
PyTorch 是一个基于 Python 的科学计算包,主要服务于两个广泛的目的:
NumPy 的替代品,用于利用 GPU 和其他加速器的能力。
一个自动微分库,对实现神经网络非常有用。
本教程的目标:¶
从高层次理解PyTorch的张量库和神经网络。
训练一个小型神经网络来分类图像
要运行下面的教程,请确保你已经安装了torch、torchvision和matplotlib包。
Tensors
在本教程中,您将学习PyTorch张量的基础知识。
A Gentle Introduction to torch.autograd
了解自动求导。
Neural Networks
本教程演示了如何在PyTorch中训练神经网络。
Training a Classifier
学习如何使用CIFAR10数据集在PyTorch中训练图像分类器。