Add openmpi to our base cuda container for MPI support (#6026)

* Add openmpi to our base container for DeepSpeed MPI support * conda Co-authored-by: Jirka Borovec <jirka.borovec@seznam.cz>
Lightning-AI · Feb 17, 2021 · 5157ba5 · 5157ba5
1 parent 99da0d9
commit 5157ba5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/dockers/base-conda/Dockerfile b/dockers/base-conda/Dockerfile
@@ -42,6 +42,7 @@ RUN apt-get update -qq && \
         curl \
         unzip \
         ca-certificates \
+        libopenmpi-dev \
     && \
 
 # Install conda and python.

diff --git a/dockers/base-cuda/Dockerfile b/dockers/base-cuda/Dockerfile
@@ -49,6 +49,7 @@ RUN apt-get update -qq && \
         unzip \
         ca-certificates \
         software-properties-common \
+        libopenmpi-dev \
     && \
 
 # Install python
-Original file line number
+Diff line change
@@ Expand Up / @@ -42,6 +42,7 @@ RUN apt-get update -qq && \ @@
             curl \
             unzip \
             ca-certificates \
+            libopenmpi-dev \
         && \
     # Install conda and python.
@@ Expand Down @@