15 lines
581 B
Docker
15 lines
581 B
Docker
FROM horovod/horovod:0.20.0-tf2.3.0-torch1.6.0-mxnet1.6.0.post0-py3.7-cuda10.1
|
|
|
|
# mpi-operator mounts the .ssh folder from a Secret. For that to work, we need
|
|
# to disable UserKnownHostsFile to avoid write permissions.
|
|
# Disabling StrictModes avoids directory and files read permission checks.
|
|
RUN echo " UserKnownHostsFile /dev/null" >> /etc/ssh/ssh_config && \
|
|
sed -i 's/#\(StrictModes \).*/\1no/g' /etc/ssh/sshd_config
|
|
|
|
RUN mkdir /tensorflow
|
|
WORKDIR "/tensorflow"
|
|
RUN git clone https://github.com/tensorflow/benchmarks
|
|
WORKDIR "/tensorflow/benchmarks"
|
|
|
|
CMD ["/bin/bash"]
|