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

Author: peterschmidt85

docker/base/base-efa.Dockerfile

@@ -8,6 +8,8 @@ ENV CUDA_PATH=/usr/local/cuda
 ENV LIBFABRIC_PATH=/opt/amazon/efa
 ENV OPEN_MPI_PATH=/opt/amazon/openmpi
 ENV NCCL_TESTS_HOME=/opt/nccl-tests
+ENV PATH="${LIBFABRIC_PATH}/bin:${OPEN_MPI_PATH}/bin:${PATH}"
+ENV LD_LIBRARY_PATH="${OPEN_MPI_PATH}/lib:${LD_LIBRARY_PATH}"
 
 # Install build dependencies
 RUN export DEBIAN_FRONTEND=noninteractive \