[UX] Pre-build a EFA version of the default Docker image #2793

Author: peterschmidt85

docker/base/Dockerfile

@@ -60,7 +60,8 @@ COPY --from=builder ${NCCL_TESTS_HOME}/build ${NCCL_TESTS_HOME}
 ARG FLAVOR
 
 # Configure library paths
-RUN apt-get install -y --no-install-recommends openmpi-bin \
+RUN apt-get update \
+    && apt-get install -y --no-install-recommends openmpi-bin \
     && if [ "$FLAVOR" = "devel" ]; then \
         cuda_version=$(echo ${CUDA_VERSION} | awk -F . '{ print $1"-"$2 }') \
         && apt-get install -y --no-install-recommends \

docker/base/efa.Dockerfile

@@ -14,7 +14,8 @@ ARG NCCL_VERSION=2.26.2-1
 ARG OFI_VERSION=1.14.0
 ARG FLAVOR
 
-RUN cuda_version=$(echo ${CUDA_VERSION} | awk -F . '{ print $1"-"$2 }') \
+RUN apt-get update \
+    && cuda_version=$(echo ${CUDA_VERSION} | awk -F . '{ print $1"-"$2 }') \
     && apt-get install -y --no-install-recommends \
         cuda-libraries-dev-${cuda_version} \
         cuda-nvcc-${cuda_version} \