• Get Started
  • Features
  • Ecosystem
  • Blog
  • Tutorials
  • Docs
  • Resources
  • Github

Ecosystem

Tap into a rich ecosystem of tools, libraries, and more to support, accelerate, and explore AI development.

AllenNLP

AllenNLP is an open-source research library built on PyTorch for designing and evaluating deep learning models for NLP.

ELF

ELF is a platform for game research that allows developers to train and test their algorithms in various game environments.

Glow

Glow is a machine learning compiler that accelerates the performance of deep learning frameworks on different hardware platforms.

GPyTorch

GPyTorch is a Gaussian process library implemented using PyTorch, designed for creating scalable, flexible Gaussian process models.

Pyro

Pyro is a universal probabilistic programming language (PPL) written in Python and supported by PyTorch on the backend.

TensorLy

TensorLy is a Python library that makes tensor learning simple and accessible.

Translate

Translate is an open source project based on Facebook's machine translation systems.

WSL Embeddings

WSL Embeddings allow developers to experiment with image recognition capabilities based on models trained on large datasets [Coming Soon]

Quick Start
Locally

Select your preferences and run the install command. Please ensure that you are on the latest pip and numpy packages. Anaconda is our recommended package manager. You can also install previous versions of PyTorch.

Your OS
Package
Python
Cuda
Run this Command:
Your OS
Linux
Mac
Windows
Package
Conda
Pip
Source
Python
2.7
3.5
3.6
3.7
Cuda
8.0
9.0
9.2
None
Run this Command:
conda install pytorch torchvision -c pytorch

Quick Start With
Cloud Partners

Get up and running with PyTorch quickly through popular cloud platforms and machine learning services.

AWS
  • Amazon SageMaker
  • AWS Deep Learning AMIs
  • Learn More
Google Cloud Platform
  • Cloud Deep Learning VM Image
Microsoft Azure
  • Azure Machine Learning Studio
  • Data Science Virtual Machines
  • Azure Notebooks
  • Learn More

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources
  • PyTorch
  • Get Started
  • Features
  • Ecosystem
  • Blog
  • Contributing
  • Resources
  • Tutorials
  • Docs
  • Discuss
  • GitHub Issues
  • Slack
  • Stay Connected
  • Get Started
  • Features
  • Ecosystem
  • Blog
  • Tutorials
  • Docs
  • Resources
  • Github