Docs

Tensors and Dynamic neural networks in Python
with strong GPU acceleration.

PyTorch is a deep learning framework that puts Python first.

We are in an early-release Beta. Expect some adventures.

Learn More
Fork me on GitHub

Get Started.

Select your preferences, then run the
PyTorch install command.

Please ensure that you are on the latest pip and numpy packages.
Anaconda is our recommended package manager

OS
Linux
OSX
Package Manager
conda
pip
Source
Python
2.7
3.5
3.6
CUDA
7.5
8.0
None
Run this command:

Example Projects

Browse projects built using PyTorch.

Explore Examples

Tutorials

Learn how to use the PyTorch library

Browse Tutorials

Companies & Universities developing PyTorch