OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 00:21:07 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
b6f43afaca
pytorch/benchmarks
History
Zafar Takhirov 0ae063d5d9 Fixed concatenation benchmark + added it to the microbenchmarking runs 
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/31587

Test Plan: Imported from OSS

Differential Revision: D19221813

Pulled By: z-a-f

fbshipit-source-id: ee0eb60da7899b23fdc63326302d1e2fd4b540ee
2020-01-03 11:23:12 -08:00
..
fastrnns Some modifications to improve readability (#31352) 2020-01-02 12:48:34 -08:00
framework_overhead_benchmark Added running via throughput benchmark options. (#23077) 2019-07-22 11:27:55 -07:00
operator_benchmark Fixed concatenation benchmark + added it to the microbenchmarking runs 2020-01-03 11:23:12 -08:00
README.md Move fast rnn benchmark to pytorch/pytorch 2019-03-27 14:46:09 -07:00

README.md

PyTorch Benchmarks

NOTE: This folder is currently work in progress.

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 supercede 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