Commit 8b0d900a authored by Abraham, Subil's avatar Abraham, Subil
Browse files

aaaaaaaaa

parent 4660cebb

tensorflowimage-centos-cuda/Dockerfile.spectrummpicommunity

+3 −1
Original line number Diff line number Diff line
@@ -148,4 +148,6 @@ RUN pip install /tmp/tensorflow_pkg/tensorflow-2.5.0-cp39-cp39-linux_ppc64le.whl 


ENV LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/opt/ibm/spectrum_mpi/lib

ENV PATH=$PATH:/opt/ibm/spectrum_mpi/bin

RUN HOROVOD_WITH_MPI=1 pip install --no-cache-dir --no-binary horovod horovod
 
ENV IBM_SPECTRUM_MPI_LICENSE_ACCEPT=yes

RUN /opt/ibm/spectrum_mpi/lap_ce/bin/accept_spectrum_mpi_license.sh

RUN HOROVOD_WITH_MPI=1 CC=mpicc CXX=mpicxx pip install --no-cache-dir --no-binary horovod horovod

tensorflowimage-centos-cuda/tmpDoc

0 → 100644
+6 −0
Original line number Diff line number Diff line 
FROM code.ornl.gov:4567/76a/olcfbaseimages/tensorflowimage-spectrummpicommunity-centos-cuda:latest



ENV IBM_SPECTRUM_MPI_LICENSE_ACCEPT=yes

RUN /opt/ibm/spectrum_mpi/lap_ce/bin/accept_spectrum_mpi_license.sh

RUN pip uninstall -y horovod

RUN HOROVOD_WITH_MPI=1 CC=mpicc CXX=mpicxx pip install --no-cache-dir --no-binary horovod horovod