Add TF1 platform, docker and README
This commit is contained in:
parent
16b7239cd7
commit
dbe5490c5b
28 changed files with 655 additions and 34 deletions
32
docker-compose.yml
Normal file
32
docker-compose.yml
Normal 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}"
|
||||
Loading…
Add table
Add a link
Reference in a new issue