Skip to content

Commit b59b3d3

Browse files
committed
fix: remove transformer-neuronx
1 parent 4963f3d commit b59b3d3

File tree

1 file changed

+11
-15
lines changed

1 file changed

+11
-15
lines changed

huggingface/pytorch/inference/docker/2.7/py3/sdk2.24.0/Dockerfile.neuronx

Lines changed: 11 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -35,32 +35,29 @@ ENV TEMP=/home/model-server/tmp
3535

3636
RUN apt-get update \
3737
&& apt-get upgrade -y \
38-
&& apt-get install -y --no-install-recommends software-properties-common \
39-
&& add-apt-repository ppa:openjdk-r/ppa \
40-
&& apt-get update \
4138
&& apt-get install -y --no-install-recommends \
42-
build-essential \
4339
apt-transport-https \
40+
build-essential \
4441
ca-certificates \
4542
cmake \
4643
curl \
4744
emacs \
4845
git \
46+
gnupg2 \
47+
gpg-agent \
4948
jq \
5049
libgl1-mesa-glx \
50+
libglib2.0-0 \
5151
libsm6 \
5252
libxext6 \
5353
libxrender-dev \
54-
libgssapi-krb5-2 \
54+
libcap-dev \
55+
libhwloc-dev \
5556
openjdk-11-jdk \
57+
unzip \
5658
vim \
5759
wget \
58-
unzip \
5960
zlib1g-dev \
60-
libcap-dev \
61-
gpg-agent \
62-
libexpat1 \
63-
libxml2 \
6461
&& rm -rf /var/lib/apt/lists/* \
6562
&& rm -rf /tmp/tmp* \
6663
&& apt-get clean
@@ -132,7 +129,6 @@ RUN pip install --extra-index-url https://pip.repos.neuron.amazonaws.com \
132129
neuronx-cc==$NEURONX_CC_VERSION \
133130
torch-neuronx==$NEURONX_FRAMEWORK_VERSION \
134131
neuronx_distributed==$NEURONX_DISTRIBUTED_VERSION \
135-
transformers-neuronx==$NEURONX_TRANSFORMERS_VERSION \
136132
&& pip install "protobuf>=3.18.3,<4" \
137133
&& pip install --no-deps --no-cache-dir -U torchvision==0.16.*
138134

@@ -192,7 +188,7 @@ RUN HOME_DIR=/root \
192188
# conda leaves an empty /root/.cache/conda/notices.cache file which is not removed by conda clean -ya
193189
&& rm -rf ${HOME_DIR}/.cache/conda
194190

195-
ENV HF_HUB_USER_AGENT_ORIGIN="aws:sagemaker:neuron:inference:regular"
196-
EXPOSE 8080 8081
197-
ENTRYPOINT ["python", "/usr/local/bin/dockerd-entrypoint.py"]
198-
CMD ["serve"]
191+
# ENV HF_HUB_USER_AGENT_ORIGIN="aws:sagemaker:neuron:inference:regular"
192+
# EXPOSE 8080 8081
193+
# ENTRYPOINT ["python", "/usr/local/bin/dockerd-entrypoint.py"]
194+
# CMD ["serve"]

0 commit comments

Comments
 (0)