13 lines
301 B
Docker
13 lines
301 B
Docker
FROM rongou/horovod
|
|
|
|
RUN mkdir /tensorflow
|
|
WORKDIR "/tensorflow"
|
|
RUN git clone https://github.com/tensorflow/benchmarks
|
|
WORKDIR "/tensorflow/benchmarks"
|
|
|
|
CMD mpirun \
|
|
python scripts/tf_cnn_benchmarks/tf_cnn_benchmarks.py \
|
|
--model resnet101 \
|
|
--batch_size 64 \
|
|
--variable_update horovod
|