@@ -6,24 +6,24 @@ ARG BASE_IMG=<jupyter>
66FROM $BASE_IMG
77
88# Content below is based on the scripts/Dockerfiles here:
9- # https://github.com/HabanaAI/Setup_and_Install/blob/1.17.1 /dockerfiles/base/Dockerfile.ubuntu22.04
10- # https://github.com/HabanaAI/Setup_and_Install/blob/1.17.1 /dockerfiles/pytorch/Dockerfile.ubuntu
9+ # https://github.com/HabanaAI/Setup_and_Install/blob/1.19.0 /dockerfiles/base/Dockerfile.ubuntu22.04
10+ # https://github.com/HabanaAI/Setup_and_Install/blob/1.19.0 /dockerfiles/pytorch/Dockerfile.ubuntu
1111
1212# args - gaudi version
13- ARG GAUDI_VERSION=1.17.1
14- ARG GAUDI_REVISION=40
13+ ARG GAUDI_VERSION=1.19.2
14+ ARG GAUDI_REVISION=32
1515
1616# args - software versions
1717# see this support matrix for compatible versions:
1818# https://docs.habana.ai/en/latest/Support_Matrix/Support_Matrix.html
19- ARG AWS_EFA_VERSION=1.29 .0
19+ ARG AWS_EFA_VERSION=1.34 .0
2020ARG HCCL_OFI_WRAPPER_VERSION=1.18.0
21- ARG LIBFABRIC_VERSION=1.20 .0
22- ARG PYTORCH_VERSION=2.3 .1
21+ ARG LIBFABRIC_VERSION=1.22 .0
22+ ARG PYTORCH_VERSION=2.5 .1
2323
24- # Gaudi 1.17 does not currently support Python 3.11, so we downgrade to 3.10
24+ # Gaudi 1.19 does not currently support Python 3.11, so we downgrade to 3.10
2525# https://docs.habana.ai/en/latest/Support_Matrix/Support_Matrix.html
26- ARG PYTHON_VERSION=3.10.14
26+ ARG PYTHON_VERSION=3.10.16
2727RUN sed -i "s/python ==.*/python ==${PYTHON_VERSION}/" ${CONDA_DIR}/conda-meta/pinned \
2828 && conda install -y -q \
2929 python==${PYTHON_VERSION} \
0 commit comments