A 60 Minute Blitz — PyTorch Tutorials 2.7.0+cu126 documentation (original) (raw)

Deep Learning with PyTorch: A 60 Minute Blitz¶

Created On: Mar 24, 2017 | Last Updated: May 31, 2023 | Last Verified: Nov 05, 2024

Author: Soumith Chintala

What is PyTorch?¶

PyTorch is a Python-based scientific computing package serving two broad purposes:

• A replacement for NumPy to use the power of GPUs and other accelerators.
• An automatic differentiation library that is useful to implement neural networks.

Goal of this tutorial:¶

• Understand PyTorch’s Tensor library and neural networks at a high level.
• Train a small neural network to classify images

To run the tutorials below, make sure you have the torch, torchvision, and matplotlib packages installed.