Skip to content

Files

Latest commit

727ee85 · Dec 23, 2024

History

History
This branch is 5085 commits behind pytorch/pytorch:main.

benchmarks

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
Dec 11, 2024
Dec 22, 2024
Dec 11, 2024
Dec 11, 2024
Dec 11, 2024
Dec 11, 2024
Dec 20, 2024
Jul 17, 2024
Dec 11, 2024
Dec 11, 2024
Dec 14, 2024
Jul 17, 2024
Dec 23, 2024
Jun 5, 2024
Dec 11, 2024
Dec 11, 2024
Dec 12, 2024
Dec 11, 2024
Dec 22, 2024
Jul 6, 2023
Jul 17, 2024
Sep 3, 2020
Jul 26, 2023

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: