Skip to content

Commit da5fedd

Browse files
malav-shastrimalavhs
andauthored
Patch HF PT Training latest(2.1) for packages expat, libarchive13, apparmor, gevent (#4357)
* patch libarchive13 for HF PT Inference neuronx * Patch libarchive13 in HF PT 1.13 neuronx * Patch libarchive13 in HF PT 2.1 training neuronx * Patch critical vulnerabilites in HF PT 1.13 GPU Training image * Patch HF PT Training latest for packages expat,libarchive13,apparmor * fix: pin huggingface_hub, gevent * fix typo * revert dlc_dev_congig.toml changes before merge * fix typo --------- Co-authored-by: malavhs <[email protected]>
1 parent e68820d commit da5fedd

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

huggingface/pytorch/training/docker/2.1/py3/cu121/Dockerfile.gpu

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,8 @@ ARG FLASH_ATTN_VERSION=2.3.6
1919
ARG MULTIPROCESS_VERSION=0.70.16
2020
ARG DILL_VERSION=0.3.8
2121
ARG NINJA_VERSION=1.11.1
22+
ARG HUGGINGFACE_HUB_VERSION=0.25.2
23+
ARG GEVENT_VERSION=24.10.1
2224
ARG PYTHON=python3
2325

2426
# TODO: Remove when the base image is updated
@@ -42,11 +44,13 @@ RUN pip install --no-cache-dir \
4244
flash-attn==${FLASH_ATTN_VERSION} \
4345
multiprocess==${MULTIPROCESS_VERSION} \
4446
dill==${DILL_VERSION} \
45-
ninja==${NINJA_VERSION}
47+
ninja==${NINJA_VERSION} \
48+
huggingface_hub==${HUGGINGFACE_HUB_VERSION} \
49+
gevent==${GEVENT_VERSION}
4650

4751
RUN apt-get update \
4852
# TODO: Remove upgrade statements once packages are updated in base image
49-
&& apt-get -y upgrade --only-upgrade systemd openssl cryptsetup libkrb5-3 \
53+
&& apt-get -y upgrade --only-upgrade systemd openssl cryptsetup libkrb5-3 expat libarchive13 apparmor \
5054
&& apt install -y git git-lfs \
5155
&& apt-get clean \
5256
&& rm -rf /var/lib/apt/lists/*

0 commit comments

Comments
 (0)