From
       Research To
Production

An open source machine learning framework that accelerates the path from research prototyping to production deployment.

Get Started

Key Features &
Capabilities

See all Features
TorchScript

TorchScript provides a seamless transition between eager mode and graph mode to accelerate the path to production.

Distributed Training

Scalable distributed training and performance optimization in research and production is enabled by the torch.distributed backend.

Tools & Libraries

A rich ecosystem of tools and libraries extends PyTorch and supports development in computer vision, NLP and more.

Cloud Partners

PyTorch is well supported on major cloud platforms, providing frictionless development and easy scaling.

Quick Start
Locally

Select your preferences and run the install command. Stable represents the most currently tested and supported version of PyTorch. This should be suitable for many users. Preview is available if you want the latest, not fully tested and supported, 1.5 builds that are generated nightly. Please ensure that you have met the prerequisites below (e.g., numpy), depending on your package manager. Anaconda is our recommended package manager since it installs all dependencies. You can also install previous versions of PyTorch. Note that LibTorch is only available for C++.

PyTorch Build
Your OS
Package
Language
CUDA
Run this Command:
PyTorch Build
Stable (1.4)
Preview (Nightly)
Your OS
Linux
Mac
Windows
Package
Conda
Pip
LibTorch
Source
Language
Python
C++ / Java
CUDA
9.2
10.1
None
Run this Command:
conda install pytorch torchvision -c pytorch

Previous versions of PyTorch

Quick Start With
Cloud Partners

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

Alibaba Cloud
Amazon Web Services
Google Cloud Platform
Microsoft Azure

Ecosystem

See all Projects

Explore a rich ecosystem of libraries, tools, and more to support development.

Captum

Captum (“comprehension” in Latin) is an open source, extensible library for model interpretability built on PyTorch.

PyTorch Geometric

PyTorch Geometric is a library for deep learning on irregular input data such as graphs, point clouds, and manifolds.

skorch

skorch is a high-level library for PyTorch that provides full scikit-learn compatibility.

Community

Join the PyTorch developer community to contribute, learn, and get your questions answered.

PyTorchDiscuss

Browse and join discussions on deep learning with PyTorch.

Slack

Discuss advanced topics. Request access: https://bit.ly/ptslack

中文文档

Docs and tutorials in Chinese, translated by the community.

파이토치(PyTorch)

Tutorials in Korean, translated by the community.

Companies & Universities
Using PyTorch

Pushing the state of the art in NLP and Multi-task learning.

Using PyTorch’s flexibility to efficiently research new algorithmic approaches.

Educating the next wave of AI Innovators using PyTorch.

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

To analyze traffic and optimize your experience, we serve cookies on this site. By clicking or navigating, you agree to allow our usage of cookies. As the current maintainers of this site, Facebook’s Cookies Policy applies. Learn more, including about available controls: Cookies Policy.