Skip to content

Commit 0b6a6b0

Browse files
authored
Gaudi RHOAI notebook image 1.19.0 update (#589)
Signed-off-by: sharvil10 <sharvil.shah@intel.com>
1 parent 92a56c2 commit 0b6a6b0

8 files changed

+53
-101
lines changed

enterprise/redhat/openshift-ai/gaudi/README.md

+4-3
Original file line numberDiff line numberDiff line change
@@ -6,9 +6,10 @@ Intel® Gaudi AI Software Tools for OpenShift AI(RedHat OpenShift Data Science/R
66

77
| Notebook Container Name | Tools | Image Name |
88
| -----------------------------| ------------- | ------------- |
9-
| Intel Gaudi Notebook Container 1.17.0-495 | [Intel® Gaudi Software Stack*](https://docs.habana.ai/en/latest/Installation_Guide/Bare_Metal_Fresh_OS.html), [Intel® Gaudi PyTorch](https://docs.habana.ai/en/latest/PyTorch/index.html), [Intel® Gaudi vLLM](https://github.com/HabanaAI/vllm-fork.git), [Intel® Gaudi DeepSpeed](https://github.com/HabanaAI/DeepSpeed) | [`registry.connect.redhat.com/intel/gaudi-notebooks:1.17.0-495-rhel-9.2`](registry.connect.redhat.com/intel/gaudi-notebooks@sha256:a62baf968caa7dd23b7f4cdcddc26e109d894f1436e247b4ea1e2fb4a5c94d54) |
10-
| Intel Gaudi Notebook Container 1.17.1-40 | [Intel® Gaudi Software Stack*](https://docs.habana.ai/en/latest/Installation_Guide/Bare_Metal_Fresh_OS.html), [Intel® Gaudi PyTorch](https://docs.habana.ai/en/latest/PyTorch/index.html), [Intel® Gaudi vLLM](https://github.com/HabanaAI/vllm-fork.git), [Intel® Gaudi DeepSpeed](https://github.com/HabanaAI/DeepSpeed) | [`registry.connect.redhat.com/intel/gaudi-notebooks:1.17.1-40-rhel-9.2`](registry.connect.redhat.com/intel/gaudi-notebooks@sha256:00ca535956b7fcdd91e71bc4a3cd4493ddcaceea9b8d7bb95a7edc0e1cb0bac4) |
11-
| Intel Gaudi Notebook Container 1.18.0-524 | [Intel® Gaudi Software Stack*](https://docs.habana.ai/en/latest/Installation_Guide/Bare_Metal_Fresh_OS.html), [Intel® Gaudi PyTorch](https://docs.habana.ai/en/latest/PyTorch/index.html), [Intel® Gaudi vLLM](https://github.com/HabanaAI/vllm-fork.git), [Intel® Gaudi DeepSpeed](https://github.com/HabanaAI/DeepSpeed) | [`registry.connect.redhat.com/intel/gaudi-notebooks:1.18.0-524-rhel-9.2`](registry.connect.redhat.com/intel/gaudi-notebooks@sha256:142b11253e5708ff9744c895868b2adda2f6f01c40127b71f1aca3d7a6e6bc29) |
9+
| Intel Gaudi Notebook Container 1.19.0-561 | [Intel® Gaudi Software Stack*](https://docs.habana.ai/en/latest/Installation_Guide/Bare_Metal_Fresh_OS.html), [Intel® Gaudi PyTorch](https://docs.habana.ai/en/latest/PyTorch/index.html), [Intel® Gaudi vLLM](https://github.com/HabanaAI/vllm-fork.git), [Intel® Gaudi DeepSpeed](https://github.com/HabanaAI/DeepSpeed) | [`registry.connect.redhat.com/intel/gaudi-notebooks:1.19.0-561-rhel-9.4`](https://catalog.redhat.com/software/containers/66e2072057f53c17d73d67e1?architecture=amd64&tag=1.18.0-524-rhel-9.2&image=676b2fd90e7869ebe618b78b&gti-tabs=registry-tokens), [`registry.connect.redhat.com/intel/gaudi-notebooks:1.19.0-561-rhel-9.2`](https://catalog.redhat.com/software/containers/66e2072057f53c17d73d67e1?architecture=amd64&tag=1.18.0-524-rhel-9.2&image=676b32a224dabcac4cf84cb3&gti-tabs=registry-tokens) |
10+
| Intel Gaudi Notebook Container 1.18.0-524 | [Intel® Gaudi Software Stack*](https://docs.habana.ai/en/latest/Installation_Guide/Bare_Metal_Fresh_OS.html), [Intel® Gaudi PyTorch](https://docs.habana.ai/en/latest/PyTorch/index.html), [Intel® Gaudi vLLM](https://github.com/HabanaAI/vllm-fork.git), [Intel® Gaudi DeepSpeed](https://github.com/HabanaAI/DeepSpeed) | [`registry.connect.redhat.com/intel/gaudi-notebooks:1.18.0-524-rhel-9.2`](https://catalog.redhat.com/software/containers/66e2072057f53c17d73d67e1?architecture=amd64&tag=1.18.0-524-rhel-9.2&image=67181ccca33f4e501721789f&gti-tabs=registry-tokens) |
11+
| Intel Gaudi Notebook Container 1.17.1-40 | [Intel® Gaudi Software Stack*](https://docs.habana.ai/en/latest/Installation_Guide/Bare_Metal_Fresh_OS.html), [Intel® Gaudi PyTorch](https://docs.habana.ai/en/latest/PyTorch/index.html), [Intel® Gaudi vLLM](https://github.com/HabanaAI/vllm-fork.git), [Intel® Gaudi DeepSpeed](https://github.com/HabanaAI/DeepSpeed) | [`registry.connect.redhat.com/intel/gaudi-notebooks:1.17.1-40-rhel-9.2`](https://catalog.redhat.com/software/containers/66e2072057f53c17d73d67e1?architecture=amd64&tag=1.18.0-524-rhel-9.2&image=66fc3bf38081186cfc015a0d&gti-tabs=registry-tokens) |
12+
| Intel Gaudi Notebook Container 1.17.0-495 | [Intel® Gaudi Software Stack*](https://docs.habana.ai/en/latest/Installation_Guide/Bare_Metal_Fresh_OS.html), [Intel® Gaudi PyTorch](https://docs.habana.ai/en/latest/PyTorch/index.html), [Intel® Gaudi vLLM](https://github.com/HabanaAI/vllm-fork.git), [Intel® Gaudi DeepSpeed](https://github.com/HabanaAI/DeepSpeed) | [`registry.connect.redhat.com/intel/gaudi-notebooks:1.17.0-495-rhel-9.2`](https://catalog.redhat.com/software/containers/66e2072057f53c17d73d67e1?architecture=amd64&tag=1.18.0-524-rhel-9.2&image=66e0ac59261d52855750518c&gti-tabs=registry-tokens) |
1213

1314
## Run Gaudi Notebook Containers
1415

enterprise/redhat/openshift-ai/gaudi/crd-sample.yaml

+4-2
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ apiVersion: aitools.intel/v1
1616
kind: GaudiAIToolsContainer
1717
metadata:
1818
labels:
19-
gaudi-software-version: '1.17.1-40'
19+
gaudi-software-version: '1.19.0-561'
2020
name: intel-gaudi
2121
spec:
2222
nameOverride: ""
@@ -25,6 +25,8 @@ spec:
2525
registry: registry.connect.redhat.com
2626
repo: intel/gaudi-notebooks
2727
tags:
28-
- gaudi_software: "1.17.1-40"
28+
- gaudi_software: "1.19.0-561"
2929
rhel_os: "9.2"
30+
- gaudi_software: "1.19.0-561"
31+
rhel_os: "9.4"
3032
namespace: redhat-ods-applications

enterprise/redhat/openshift-ai/gaudi/docker/Dockerfile.rhel9.2

+4-8
Original file line numberDiff line numberDiff line change
@@ -69,6 +69,7 @@ RUN mkdir -p /licenses && \
6969
ENV PYTHON_VERSION=3.10
7070
COPY install-python310.sh .
7171
RUN ./install-python310.sh rhel9.2 && rm install-python310.sh
72+
RUN echo "/usr/local/lib" > /etc/ld.so.conf.d/python.conf && ldconfig
7273
ENV LD_LIBRARY_PATH=/usr/local/lib:$LD_LIBRARY_PATH
7374

7475
COPY install_efa.sh .
@@ -194,17 +195,11 @@ RUN dnf install --allowerasing -y \
194195
gperftools-devel && \
195196
dnf clean all && rm -rf /var/cache/yum
196197

197-
RUN dnf config-manager --add-repo https://yum.repos.intel.com/mkl/setup/intel-mkl.repo -y && \
198-
dnf install --allowerasing -y intel-mkl-64bit-2020.4-912 && \
199-
dnf clean all && rm -rf /var/cache/yum
200-
201198
# Set LD_PRELOAD after all required installations to
202199
# avoid warnings during docker creation
203200
ENV LD_PRELOAD=/lib64/libtcmalloc.so.4
204201
ENV TCMALLOC_LARGE_ALLOC_REPORT_THRESHOLD=7516192768
205202

206-
RUN rm -rf /tmp/*
207-
208203
USER 1001
209204

210205
COPY --chown=1001:0 install_packages.sh .
@@ -238,9 +233,10 @@ RUN python -m pip install -r requirements.txt && \
238233
RUN cd ${APP_ROOT}/ && \
239234
git clone https://github.com/HabanaAI/vllm-fork.git && \
240235
cd vllm-fork && \
241-
VLLM_TARGET_DEVICE=hpu pip install -e .
236+
git checkout habana_main && \
237+
pip install -r requirements-hpu.txt && \
238+
VLLM_TARGET_DEVICE=hpu python setup.py develop
242239

243240
WORKDIR ${APP_ROOT}/src
244241
ENV NOTEBOOK_SAMPLES_LINK="https://raw.githubusercontent.com/intel/ai-containers/refs/heads/main/enterprise/redhat/openshift-ai/gaudi/demo/oneapi-sample.ipynb"
245-
246242
ENTRYPOINT ["bash", "-c", "/opt/app-root/builder/run"]

enterprise/redhat/openshift-ai/gaudi/docker/Dockerfile.rhel9.4

+27-43
Original file line numberDiff line numberDiff line change
@@ -29,10 +29,15 @@ RUN echo "[CRB]" > /etc/yum.repos.d/CentOS-Linux-CRB.repo && \
2929
echo "gpgkey=https://www.centos.org/keys/RPM-GPG-KEY-CentOS-Official-SHA256" >> /etc/yum.repos.d/CentOS-Linux-CRB.repo && \
3030
echo "gpgcheck=1" >> /etc/yum.repos.d/CentOS-Linux-CRB.repo
3131

32-
RUN dnf install -y https://dl.fedoraproject.org/pub/epel/epel-release-latest-9.noarch.rpm && \
33-
dnf clean all && rm -rf /var/cache/yum
34-
3532
RUN dnf install -y \
33+
python3-dnf-plugin-versionlock && \
34+
dnf versionlock add redhat-release* && \
35+
dnf clean all
36+
37+
RUN dnf update -y && dnf install -y https://dl.fedoraproject.org/pub/epel/epel-release-latest-9.noarch.rpm && \
38+
dnf clean all
39+
40+
RUN dnf update -y && dnf install -y \
3641
clang \
3742
cmake3 \
3843
cpp \
@@ -49,8 +54,8 @@ RUN dnf install -y \
4954
lsof \
5055
python3-devel \
5156
openssh-clients \
52-
openssl-1:3.0.7-28.el9_4 \
53-
openssl-devel-1:3.0.7-28.el9_4 \
57+
openssl \
58+
openssl-devel \
5459
libjpeg-devel \
5560
openssh-server \
5661
lsb_release \
@@ -66,24 +71,15 @@ RUN dnf install -y \
6671
python3.11-pip \
6772
python3.11-devel \
6873
python3.11-rpm \
69-
ffmpeg-free \
70-
python3-dnf-plugin-versionlock && \
74+
ffmpeg-free && \
7175
# update pkgs (except OS version) for resolving potentials CVEs
72-
dnf versionlock add redhat-release* openssl* libcurl-minimal curl-minimal ima-evm-utils python3-rpm rpm* && \
73-
dnf update -y && \
76+
dnf versionlock add python3-rpm rpm* && \
7477
dnf clean all && rm -rf /var/cache/yum && \
7578
rm -f /etc/ssh/ssh_host_*_key*
7679

7780
RUN mkdir -p /licenses && \
7881
wget -O /licenses/LICENSE https://raw.githubusercontent.com/intel/ai-containers/main/LICENSE
7982

80-
RUN alternatives --install /usr/bin/python3 python3 /usr/bin/python3.11 2 && \
81-
alternatives --install /usr/bin/python3 python3 /usr/bin/python3.9 1 && \
82-
alternatives --set python3 /usr/bin/python3.11 && \
83-
alternatives --install /usr/bin/pip3 pip3 /usr/bin/pip3.11 2 && \
84-
alternatives --install /usr/bin/pip3 pip3 /usr/bin/pip3.9 1 && \
85-
alternatives --set pip3 /usr/bin/pip3.11
86-
8783
COPY install_efa.sh .
8884
RUN ./install_efa.sh && rm install_efa.sh && rm -rf /etc/ld.so.conf.d/efa.conf /etc/profile.d/efa.sh
8985

@@ -193,32 +189,18 @@ RUN echo "[CRB]" > /etc/yum.repos.d/CentOS-Linux-CRB.repo && \
193189
echo "gpgkey=https://www.centos.org/keys/RPM-GPG-KEY-CentOS-Official-SHA256" >> /etc/yum.repos.d/CentOS-Linux-CRB.repo && \
194190
echo "gpgcheck=1" >> /etc/yum.repos.d/CentOS-Linux-CRB.repo
195191

196-
RUN dnf install --allowerasing -y \
197-
curl-7.76.1-29.el9_4.1 \
198-
cairo-devel \
199-
numactl-devel \
200-
iproute \
201-
which \
202-
zlib-devel \
203-
lapack-devel \
204-
openblas-devel \
205-
numactl \
206-
gperftools-devel && \
207-
dnf clean all && rm -rf /var/cache/yum
208-
209-
RUN echo "[oneAPI]" >> /etc/yum.repos.d/oneAPI.repo && \
210-
echo "name=Intel® oneAPI repository" >> /etc/yum.repos.d/oneAPI.repo && \
211-
echo "baseurl=https://yum.repos.intel.com/oneapi" >> /etc/yum.repos.d/oneAPI.repo && \
212-
echo 'enabled=1' >> /etc/yum.repos.d/oneAPI.repo && \
213-
echo "gpgcheck=1" >> /etc/yum.repos.d/oneAPI.repo && \
214-
echo "repo_gpgcheck=1" >> /etc/yum.repos.d/oneAPI.repo && \
215-
echo "gpgkey=https://yum.repos.intel.com/intel-gpg-keys/GPG-PUB-KEY-INTEL-SW-PRODUCTS.PUB" >> /etc/yum.repos.d/oneAPI.repo
216-
217-
RUN dnf install --allowerasing -y intel-oneapi-mkl-2024.2.0 && \
192+
RUN dnf update -y && dnf install --nodocs --setopt=install_weak_deps=false --allowerasing -y \
193+
cairo-devel \
194+
numactl-devel \
195+
iproute \
196+
which \
197+
zlib-devel \
198+
lapack-devel \
199+
openblas-devel \
200+
numactl \
201+
gperftools-devel && \
218202
dnf clean all && rm -rf /var/cache/yum
219203

220-
ENV LD_LIBRARY_PATH=/opt/intel/oneapi/mkl/2024.2/lib:${LD_LIBRARY_PATH}
221-
222204
RUN rm -rf /tmp/*
223205

224206
USER 1001
@@ -260,9 +242,11 @@ RUN python -m pip install -r requirements.txt && \
260242
RUN cd ${APP_ROOT}/ && \
261243
git clone https://github.com/HabanaAI/vllm-fork.git && \
262244
cd vllm-fork && \
263-
VLLM_TARGET_DEVICE=hpu pip install -e .
245+
git checkout habana_main && \
246+
pip install -r requirements-hpu.txt && \
247+
VLLM_TARGET_DEVICE=hpu python setup.py develop
264248

265249
WORKDIR ${APP_ROOT}/src
266-
ENV JUPYTER_PRELOAD_REPOS="https://github.com/IntelAI/oneAPI-samples"
267-
ENV REPO_BRANCH="main"
250+
ENV NOTEBOOK_SAMPLES_LINK="https://raw.githubusercontent.com/intel/ai-containers/refs/heads/main/enterprise/redhat/openshift-ai/gaudi/demo/oneapi-sample.ipynb"
251+
268252
ENTRYPOINT ["bash", "-c", "/opt/app-root/builder/run"]

enterprise/redhat/openshift-ai/gaudi/docker/docker-compose.yaml

+11-11
Original file line numberDiff line numberDiff line change
@@ -22,12 +22,12 @@ services:
2222
https_proxy: ${https_proxy}
2323
no_proxy: ""
2424
ARTIFACTORY_URL: ${ARTIFACTORY_URL:-vault.habana.ai}
25-
VERSION: ${VERSION:-1.18.0}
26-
REVISION: ${REVISION:-524}
25+
VERSION: ${VERSION:-1.19.0}
26+
REVISION: ${REVISION:-561}
2727
context: .
2828
target: gaudi-base
2929
dockerfile: Dockerfile.rhel${RHEL_OS:-9.2}
30-
image: ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-gaudi-base-${VERSION:-1.18.0}-${REVISION:-524}-rhel-${RHEL_OS:-9.2}
30+
image: ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-gaudi-base-${VERSION:-1.19.0}-${REVISION:-561}-rhel-${RHEL_OS:-9.2}
3131
entrypoint: ["/bin/bash", "-c"]
3232
command: >
3333
"hl-smi"
@@ -37,17 +37,17 @@ services:
3737
BASE_IMAGE: ${BASE_IMAGE:-registry.access.redhat.com/ubi9/ubi}
3838
BASE_TAG: ${RHEL_OS:-9.2}
3939
BASE_NAME: rhel${RHEL_OS:-rhel9.2}
40-
PT_VERSION: ${PT_VERSION:-2.4.0}
40+
PT_VERSION: ${PT_VERSION:-2.5.1}
4141
http_proxy: ${http_proxy}
4242
https_proxy: ${https_proxy}
4343
no_proxy: ""
4444
ARTIFACTORY_URL: ${ARTIFACTORY_URL:-vault.habana.ai}
45-
VERSION: ${VERSION:-1.18.0}
46-
REVISION: ${REVISION:-524}
45+
VERSION: ${VERSION:-1.19.0}
46+
REVISION: ${REVISION:-561}
4747
context: .
4848
target: gaudi-pytorch
4949
dockerfile: Dockerfile.rhel${RHEL_OS:-9.2}
50-
image: ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-gaudi-pytorch-${VERSION:-1.18.0}-${REVISION:-524}-rhel-${RHEL_OS:-9.2}
50+
image: ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-gaudi-pytorch-${VERSION:-1.19.0}-${REVISION:-561}-rhel-${RHEL_OS:-9.2}
5151
entrypoint: ["/bin/bash", "-c"]
5252
command: >
5353
"python -c 'import torch'"
@@ -57,17 +57,17 @@ services:
5757
BASE_IMAGE: ${BASE_IMAGE:-registry.access.redhat.com/ubi9/ubi}
5858
BASE_TAG: ${RHEL_OS:-9.2}
5959
BASE_NAME: ${BASE_NAME:-rhel9.2}
60-
PT_VERSION: ${PT_VERSION:-2.4.0}
60+
PT_VERSION: ${PT_VERSION:-2.5.1}
6161
http_proxy: ${http_proxy}
6262
https_proxy: ${https_proxy}
6363
no_proxy: ""
6464
ARTIFACTORY_URL: ${ARTIFACTORY_URL:-vault.habana.ai}
65-
VERSION: ${VERSION:-1.18.0}
66-
REVISION: ${REVISION:-524}
65+
VERSION: ${VERSION:-1.19.0}
66+
REVISION: ${REVISION:-561}
6767
context: .
6868
target: gaudi-notebooks
6969
dockerfile: Dockerfile.rhel${RHEL_OS:-9.2}
70-
image: ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-gaudi-notebook-${VERSION:-1.18.0}-${REVISION:-524}-rhel-${RHEL_OS:-9.2}
70+
image: ${REGISTRY}/${REPO}:b-${GITHUB_RUN_NUMBER:-0}-gaudi-notebook-${VERSION:-1.19.0}-${REVISION:-561}-rhel-${RHEL_OS:-9.2}
7171
entrypoint: ["/bin/bash", "-c"]
7272
command: >
7373
"python -m jupyter notebook --version"

enterprise/redhat/openshift-ai/gaudi/docker/install-python310.sh

+2-30
Original file line numberDiff line numberDiff line change
@@ -24,12 +24,12 @@ case "${_BASE_NAME}" in
2424
echo "Skip install Python3.10 from source on Ubuntu22.04"
2525
exit 0
2626
;;
27-
*debian* | *ubuntu*)
27+
*ubuntu*)
2828
apt update
2929
apt install -y libsqlite3-dev libreadline-dev
3030
;;
3131
*rhel*)
32-
yum install -y sqlite-devel readline-devel xz-devel
32+
dnf install -y sqlite-devel readline-devel xz-devel
3333
;;
3434
*tencentos3.1*)
3535
dnf install -y sqlite-devel readline-devel zlib-devel xz-devel bzip2-devel libffi-devel
@@ -42,21 +42,6 @@ case "${_BASE_NAME}" in
4242
make && make install
4343
ln -s /etc/pki/tls/cert.pem /usr/local/openssl-1.1.1w/ssl/cert.pem
4444

45-
PATH=$PATH:/usr/local/protoc/bin:/usr/local/openssl-1.1.1w/bin
46-
LD_LIBRARY_PATH=/usr/local/openssl-1.1.1w/lib:$LD_LIBRARY_PATH
47-
_SSL_LIB="--with-openssl=/usr/local/openssl-1.1.1w"
48-
;;
49-
*amzn2*)
50-
yum install -y sqlite-devel readline-devel
51-
wget -nv -O /opt/openssl-1.1.1w.tar.gz https://github.com/openssl/openssl/releases/download/OpenSSL_1_1_1w/openssl-1.1.1w.tar.gz &&
52-
cd /opt/ &&
53-
tar xzf openssl-1.1.1w.tar.gz &&
54-
rm -rf openssl-1.1.1w.tar.gz &&
55-
cd openssl-1.1.1w &&
56-
./config --prefix=/usr/local/openssl-1.1.1w shared zlib &&
57-
make && make install
58-
ln -s /etc/pki/tls/cert.pem /usr/local/openssl-1.1.1w/ssl/cert.pem
59-
6045
PATH=$PATH:/usr/local/protoc/bin:/usr/local/openssl-1.1.1w/bin
6146
LD_LIBRARY_PATH=/usr/local/openssl-1.1.1w/lib:$LD_LIBRARY_PATH
6247
_SSL_LIB="--with-openssl=/usr/local/openssl-1.1.1w"
@@ -75,9 +60,6 @@ make -j && make altinstall
7560
# post install
7661
case "${_BASE_NAME}" in
7762
*rhel9*)
78-
alternatives --install /usr/bin/python3 python3 /usr/local/bin/python3.10 2 &&
79-
alternatives --install /usr/bin/python3 python3 /usr/bin/python3.9 1 &&
80-
alternatives --set python3 /usr/local/bin/python3.10
8163
export LD_LIBRARY_PATH=/usr/local/lib:$LD_LIBRARY_PATH
8264
;;
8365
*tencentos3.1*)
@@ -88,16 +70,6 @@ case "${_BASE_NAME}" in
8870
alternatives --set python3 /usr/local/bin/python3.10
8971
export LD_LIBRARY_PATH=/usr/local/lib:$LD_LIBRARY_PATH
9072
;;
91-
*amzn2*)
92-
update-alternatives --install /usr/bin/python3 python3 /usr/local/bin/python3.10 3 &&
93-
update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.8 2 &&
94-
update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.7 1
95-
;;
96-
*debian*)
97-
update-alternatives --install /usr/bin/python3 python3 /usr/local/bin/python3.10 3
98-
update-alternatives --install /usr/bin/python3 python3 /usr/local/bin/python3.8 2
99-
update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.7 1
100-
;;
10173
esac
10274

10375
python3 -m pip install --upgrade pip setuptools

enterprise/redhat/openshift-ai/gaudi/docker/install_packages.sh

-3
Original file line numberDiff line numberDiff line change
@@ -27,9 +27,6 @@ case "${BASE_NAME}" in
2727
*rhel8*)
2828
os_string="rhel86"
2929
;;
30-
*amzn2*)
31-
os_string="amzn2"
32-
;;
3330
*tencentos*)
3431
os_string="tencentos31"
3532
;;

enterprise/redhat/openshift-ai/gaudi/docker/requirements.txt

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
# LLM Packages
2-
deepspeed @ git+https://github.com/HabanaAI/DeepSpeed.git@1.17.1
2+
deepspeed @ git+https://github.com/HabanaAI/DeepSpeed.git@1.19.0
33

44
# Datascience and useful extensions
55
kafka-python~=2.0.2

0 commit comments

Comments
 (0)