Skip to content

Files

This branch is 5 commits behind pytorch/pytorch:main.

benchmarks

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
Mar 29, 2025
May 16, 2025
Feb 16, 2025
Dec 11, 2024
Apr 27, 2025
Dec 11, 2024
Feb 27, 2025
Apr 28, 2025
Jul 17, 2024
Feb 27, 2025
Dec 11, 2024
May 15, 2025
Jul 17, 2024
Dec 23, 2024
Jun 5, 2024
Dec 11, 2024
Apr 30, 2025
May 5, 2025
Apr 27, 2025
Mar 7, 2025
Jul 6, 2023
Jul 17, 2024
Sep 3, 2020
Apr 27, 2025

PyTorch Benchmarks

This folder contains scripts that produce reproducible timings of various PyTorch features.

It also provides mechanisms to compare PyTorch with other frameworks.

Setup environment

Make sure you're on a machine with CUDA, torchvision, and pytorch installed. Install in the following order:

# Install torchvision. It comes with the pytorch stable release binary
conda install pytorch torchvision -c pytorch

# Install the latest pytorch master from source.
# It should supersede the installation from the release binary.
cd $PYTORCH_HOME
python setup.py build develop

# Check the pytorch installation version
python -c "import torch; print(torch.__version__)"

Benchmark List

Please refer to each subfolder to discover each benchmark suite. Links are provided where descriptions exist: