docker run --runtime=nvidia --rm -it nvcr.io/nvidia/tensorflow: 18.07-py3
mpiexec --allow-run-as-root --bind-to socket -np 16 python /opt/tensorflow/nvidia-examples/cnn/resnet.py --layers=50 --precision=fp16 --batch_size=512