1*da0073e9SAndroid Build Coastguard Worker# PyTorch Benchmarks 2*da0073e9SAndroid Build Coastguard Worker 3*da0073e9SAndroid Build Coastguard WorkerThis folder contains scripts that produce reproducible timings of various PyTorch features. 4*da0073e9SAndroid Build Coastguard Worker 5*da0073e9SAndroid Build Coastguard WorkerIt also provides mechanisms to compare PyTorch with other frameworks. 6*da0073e9SAndroid Build Coastguard Worker 7*da0073e9SAndroid Build Coastguard Worker## Setup environment 8*da0073e9SAndroid Build Coastguard WorkerMake sure you're on a machine with CUDA, torchvision, and pytorch installed. Install in the following order: 9*da0073e9SAndroid Build Coastguard Worker``` 10*da0073e9SAndroid Build Coastguard Worker# Install torchvision. It comes with the pytorch stable release binary 11*da0073e9SAndroid Build Coastguard Workerconda install pytorch torchvision -c pytorch 12*da0073e9SAndroid Build Coastguard Worker 13*da0073e9SAndroid Build Coastguard Worker# Install the latest pytorch master from source. 14*da0073e9SAndroid Build Coastguard Worker# It should supersede the installation from the release binary. 15*da0073e9SAndroid Build Coastguard Workercd $PYTORCH_HOME 16*da0073e9SAndroid Build Coastguard Workerpython setup.py build develop 17*da0073e9SAndroid Build Coastguard Worker 18*da0073e9SAndroid Build Coastguard Worker# Check the pytorch installation version 19*da0073e9SAndroid Build Coastguard Workerpython -c "import torch; print(torch.__version__)" 20*da0073e9SAndroid Build Coastguard Worker``` 21*da0073e9SAndroid Build Coastguard Worker 22*da0073e9SAndroid Build Coastguard Worker## Benchmark List 23*da0073e9SAndroid Build Coastguard Worker 24*da0073e9SAndroid Build Coastguard WorkerPlease refer to each subfolder to discover each benchmark suite. Links are provided where descriptions exist: 25*da0073e9SAndroid Build Coastguard Worker 26*da0073e9SAndroid Build Coastguard Worker* [Fast RNNs](fastrnns/README.md) 27*da0073e9SAndroid Build Coastguard Worker* [Dynamo](dynamo/README.md) 28*da0073e9SAndroid Build Coastguard Worker* [Functional autograd](functional_autograd_benchmark/README.md) 29*da0073e9SAndroid Build Coastguard Worker* [Instruction counts](instruction_counts/README.md) 30*da0073e9SAndroid Build Coastguard Worker* [Operator](operator_benchmark/README.md) 31*da0073e9SAndroid Build Coastguard Worker* [Overrides](overrides_benchmark/README.md) 32*da0073e9SAndroid Build Coastguard Worker* [Sparse](sparse/README.md) 33*da0073e9SAndroid Build Coastguard Worker* [Tensor expression](tensorexpr/HowToRun.md) 34