Get Started: Install PyTorch Locally or Launch Instantly on Supported Cloud Platforms
Overview This blog walks through two crucial DeepSpeed updates: (1) a PyTorch-identical backward API that enables efficient training of multimodal, multi-component models (including non-scalar backward calls), and (2) low-precision model…
Read More
Introduction As introduced in a previous blog post, Helion is a high-level DSL that empowers developers to write high-performance ML kernels using a familiar PyTorch-like syntax, delegating the complex task…
Read More
Foundation welcomes Clockwork.io, Emmi AI, NIPA, Nota AI., Yasp, CommonAI CIC, Carnegie Mellon University, Monash University, and University of Leicester Summary The PyTorch Foundation announced nine new members, adding five…
Read More
Join PyTorch Foundation
As a member of the PyTorch Foundation, you’ll have access to resources that allow you to be stewards of stable, secure, and long-lasting codebases. You can collaborate on training, local and regional events, open-source developer tooling, academic research, and guides to help new users and contributors have a productive experience.
Key Features & Capabilities
Production Ready
Transition seamlessly between eager and graph modes with TorchScript, and accelerate the path to production with TorchServe.
Distributed Training
Scalable distributed training and performance optimization in research and production is enabled by the torch.distributed backend.
Robust Ecosystem
A rich ecosystem of tools and libraries extends PyTorch and supports development in computer vision, NLP and more.
Cloud Support
PyTorch is well supported on major cloud platforms, providing frictionless development and easy scaling.
Install PyTorch
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, builds that are generated nightly. Please ensure that you have met the prerequisites below (e.g., numpy), depending on your package manager. You can also install previous versions of PyTorch. Note that LibTorch is only available for C++.
NOTE: Latest Stable PyTorch requires Python 3.10 or later.
PyTorch Build
Your OS
Package
Language
Compute Platform
Run this Command:
PyTorch Build
Stable (2.7.0)
Preview (Nightly)
Your OS
Linux
Mac
Windows
Package
Pip
LibTorch
Source
Language
Python
C++ / Java
Compute Platform
CUDA 11.8
CUDA 12.6
CUDA 12.8
ROCm 6.3
CPU
Run this Command:
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
Quick Start With Cloud Partners
Get up and running with PyTorch quickly through popular cloud platforms and machine learning services.
Ecosystem
Featured 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.
Companies & Universities Using PyTorch
Amazon Advertising
Reduce inference costs by 71% and scale out using PyTorch, TorchServe, and AWS Inferentia.
Salesforce
Pushing the state of the art in NLP and Multi-task learning.
Stanford University
Using PyTorch’s flexibility to efficiently research new algorithmic approaches.