Dockerfile: Add Package - Install openmpi for ROCm images (#181)

**Description**
Install openmpi-4.0.0 for ROCm images.

dockerfile/cuda11.1.1.dockerfile

@@ -2,6 +2,7 @@ FROM nvcr.io/nvidia/pytorch:20.12-py3
 
 # OS:
 #   - Ubuntu: 20.04
+#   - OpenMPI: 4.0.5
 #   - Docker Client: 20.10.8
 # NVIDIA:
 #   - CUDA: 11.1.1

dockerfile/rocm4.0-pytorch1.7.0.dockerfile

@@ -2,6 +2,7 @@ FROM rocm/pytorch:rocm4.0_ubuntu18.04_py3.6_pytorch_1.7.0
 
 # OS:
 #   - Ubuntu: 18.04
+#   - OpenMPI: 4.0.5
 #   - Docker Client: 20.10.8
 # AMD:
 #   - ROCm: 4.0
@@ -49,6 +50,18 @@ RUN cd /tmp && \
     tar --extract --file docker.tgz --strip-components 1 --directory /usr/local/bin/ && \
     rm docker.tgz
 
+# Install OpenMPI
+ENV OPENMPI_VERSION=4.0.5
+RUN cd /tmp && \
+    wget -q https://www.open-mpi.org/software/ompi/v4.0/downloads/openmpi-${OPENMPI_VERSION}.tar.gz && \
+    tar xzf openmpi-${OPENMPI_VERSION}.tar.gz && \
+    cd openmpi-${OPENMPI_VERSION} && \
+    ./configure --enable-orterun-prefix-by-default && \
+    make -j $(nproc) all && \
+    make install && \
+    ldconfig && \
+    rm -rf /tmp/openmpi-${OPENMPI_VERSION}*
+
 # Configure SSH
 RUN mkdir -p /root/.ssh && \
     touch /root/.ssh/authorized_keys && \

dockerfile/rocm4.2-pytorch1.7.0.dockerfile

@@ -2,6 +2,7 @@ FROM rocm/pytorch:rocm4.2_ubuntu18.04_py3.6_pytorch_1.7.0
 
 # OS:
 #   - Ubuntu: 18.04
+#   - OpenMPI: 4.0.5
 #   - Docker Client: 20.10.8
 # AMD:
 #   - ROCm: 4.2
@@ -49,6 +50,18 @@ RUN cd /tmp && \
     tar --extract --file docker.tgz --strip-components 1 --directory /usr/local/bin/ && \
     rm docker.tgz
 
+# Install OpenMPI
+ENV OPENMPI_VERSION=4.0.5
+RUN cd /tmp && \
+    wget -q https://www.open-mpi.org/software/ompi/v4.0/downloads/openmpi-${OPENMPI_VERSION}.tar.gz && \
+    tar xzf openmpi-${OPENMPI_VERSION}.tar.gz && \
+    cd openmpi-${OPENMPI_VERSION} && \
+    ./configure --enable-orterun-prefix-by-default && \
+    make -j $(nproc) all && \
+    make install && \
+    ldconfig && \
+    rm -rf /tmp/openmpi-${OPENMPI_VERSION}*
+
 # Configure SSH
 RUN mkdir -p /root/.ssh && \
     touch /root/.ssh/authorized_keys && \