PyTorch 深度学习:60分钟入门¶
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.
In this tutorial, you will learn the basics of PyTorch tensors.
Learn about autograd.
This tutorial demonstrates how you can train neural networks in PyTorch.
Learn how to train an image classifier in PyTorch by using the CIFAR10 dataset.