Skip to content

Commit

Permalink
Update 0.tensorflow.Dockerfile (#374)
Browse files Browse the repository at this point in the history
* Update 0.tensorflow.Dockerfile

There are two typos that break this image. 
Fixed variables:

ARG AWS_OFI_NCCL_VERSION=v1.8.1-aws
ARG NCCL_VERSION=v2.18.6-1

and remove the redundant 'v' during NCCL branch switching,

* Update 3.test_cases/7.tensorflow-distributed/0.tensorflow.Dockerfile

Co-authored-by: mhuguesaws <71357145+mhuguesaws@users.noreply.github.com>

* Update 3.test_cases/7.tensorflow-distributed/0.tensorflow.Dockerfile

Co-authored-by: mhuguesaws <71357145+mhuguesaws@users.noreply.github.com>

* Update 3.test_cases/7.tensorflow-distributed/0.tensorflow.Dockerfile

Co-authored-by: mhuguesaws <71357145+mhuguesaws@users.noreply.github.com>

* Update 3.test_cases/7.tensorflow-distributed/0.tensorflow.Dockerfile

Co-authored-by: mhuguesaws <71357145+mhuguesaws@users.noreply.github.com>

---------

Co-authored-by: Keita Watanabe <keitaw09@gmail.com>
Co-authored-by: mhuguesaws <71357145+mhuguesaws@users.noreply.github.com>
  • Loading branch information
3 people authored Aug 3, 2024
1 parent 80e4e5e commit 48d185e
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions 3.test_cases/7.tensorflow-distributed/0.tensorflow.Dockerfile
Original file line number Diff line number Diff line change
@@ -1,9 +1,9 @@
FROM nvcr.io/nvidia/tensorflow:23.10-tf2-py3

ARG EFA_INSTALLER_VERSION=1.30.0
ARG AWS_OFI_NCCL_VERSION=1.8.1-aws
ARG NCCL_TESTS_VERSION=master
ARG NCCL_VERSION=v2.18.6-1
ARG EFA_INSTALLER_VERSION=1.33.0
ARG AWS_OFI_NCCL_VERSION=v1.9.2-aws
ARG NCCL_TESTS_VERSION=v2.13.9
ARG NCCL_VERSION=v2.21.3-1
ARG OPEN_MPI_PATH=/opt/amazon/openmpi

RUN apt-get update -y
Expand Down Expand Up @@ -53,7 +53,7 @@ RUN cd $HOME \
###################################################
## Install NCCL
RUN cd /tmp \
&& git clone https://github.com/NVIDIA/nccl.git -b v${NCCL_VERSION} \
&& git clone https://github.com/NVIDIA/nccl.git -b ${NCCL_VERSION} \
&& cd nccl \
&& make -j src.build BUILDDIR=/usr/local \
# nvcc to target p5 and p4 instances
Expand Down

0 comments on commit 48d185e

Please sign in to comment.