PyTorch

JOIN US

PyTorch Conference North America
October 20-21, 2026
San Jose, CA
#PyTorchCon

REGISTER NOWSPONSOR

Get Started:  Install PyTorch Locally or Launch Instantly on Supported Cloud Platforms

September 17, 2026 in Announcements, Blog

PyTorch Day Japan 2026 Comes to Tokyo on December 10

PyTorch Day Japan 2026 will bring the open source AI community together in Tokyo on December 10 for a full day of technical talks and interactive discussions designed to foster…
Read More
September 16, 2026 in Blog

Open Research, Tooling & Optimization at PyTorch Conference North America 2026

TL;DR Taking place October 20 to 21 in San Jose, California, PyTorch Conference North America 2026 highlights open research, tooling, and performance optimization across compiler architecture, cross-hardware kernel domain-specific languages,…
Read More
September 16, 2026 in Blog

Low Precision Flash Attention 4: End-to-End Block-Scaled Attention for Blackwell

TL;DR We extend FlashAttention-4 with MXFP8 forward and backward, reaching 2.85 PF/s forward and 2 PF/s backward on LLM shapes. On our internal shapes, FA4 MX8 reaches 2.54 PF/s forward…
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.

EXPLORE BENEFITS

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
Previous versions of PyTorch

Could not find the right platform for your hardware? See the PyTorch Additional Platforms page.

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.

READ CASE STUDIES
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.