Deep Learning with PyTorch: A 60 Minute Blitz

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

Note

Make sure you have the torch and torchvision packages installed.

Tensors

A Gentle Introduction to torch.autograd

Neural Networks

Training a Classifier