Add TF1 platform, docker and README

This commit is contained in:
Corentin 2021-10-05 10:51:18 +09:00
commit dbe5490c5b
28 changed files with 655 additions and 34 deletions

32
docker-compose.yml Normal file
View file

@ -0,0 +1,32 @@
version: "3.8"
services:
tensorflow_v1:
build:
context: docker
dockerfile: tensorflow_v1.dockerfile
image: ayo/tf_v1_benchmark
container_name: tf_v1_benchmark
runtime: nvidia
volumes:
- .:/work
command: bash -c "umask 0000 && python benchmark.py --platform TF1 ${BENCH_ARGS}"
tensorflow:
build:
context: docker
dockerfile: tensorflow.dockerfile
image: ayo/tf_benchmark
container_name: tf_benchmark
runtime: nvidia
volumes:
- .:/work
command: bash -c "umask 0000 && python benchmark.py --platform TF2 TF2_V1 ${BENCH_ARGS}"
torch:
build:
context: docker
dockerfile: pytorch.dockerfile
image: ayo/torch_benchmark
container_name: torch_benchmark
runtime: nvidia
volumes:
- .:/work
command: bash -c "umask 0000 && python benchmark.py --platform Torch ${BENCH_ARGS}"