Skip to content

Commit 1124baa

Browse files
authored
feat: update gaudi notebooks to version 1.19.2 (kubeflow/kubeflow#7680)
Signed-off-by: Tuomas Katila <tuomas.katila@intel.com>
1 parent 8999e69 commit 1124baa

File tree

1 file changed

+9
-9
lines changed
  • components/example-notebook-servers/jupyter-pytorch-gaudi

1 file changed

+9
-9
lines changed

components/example-notebook-servers/jupyter-pytorch-gaudi/Dockerfile

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -6,24 +6,24 @@ ARG BASE_IMG=<jupyter>
66
FROM $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
2020
ARG 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
2727
RUN sed -i "s/python ==.*/python ==${PYTHON_VERSION}/" ${CONDA_DIR}/conda-meta/pinned \
2828
&& conda install -y -q \
2929
python==${PYTHON_VERSION} \

0 commit comments

Comments
 (0)