Add Dockerfile for building torchbench docker (#986)
Summary:
xwang233
Build torchbench docker based on pytorch nightly docker.
Environment:
- python 3.8
- cuda 11.3
- pytorch latest nightly
This docker supports nvidia gpu passthrough, but users need to make sure their docker daemon is running with nvidia runtime.
It also bundles torchdynamo and functorch. We will add bundling to Torch-TensorRT and TensorRT later.
The command to build:
```
$ docker pull ghcr.io/pytorch/pytorch-nightly:latest
$ pushd docker
$ docker build . -t pytorch/benchmark:latest
```
The command to run:
```
$ docker run -it --gpus all pytorch/benchmark:latest /bin/bash
(torchbench) root@5e76c95cb424:/workspace/benchmark# cd /workspace/benchmark
(torchbench) root@5e76c95cb424:/workspace/benchmark# python run.py resnet18 -d cuda
Running eval method from resnet18 on cuda in eager mode.
GPU Time: 4.578 milliseconds
CPU Total Wall Time: 4.613 milliseconds
```
Fixes https://github.com/pytorch/benchmark/issues/984
Pull Request resolved: https://github.com/pytorch/benchmark/pull/986
Reviewed By: erichan1
Differential Revision: D37353756
Pulled By: xuzhao9
fbshipit-source-id: 25197229b2915e0e5a031810a1d166aeffc4e4d7
