Skip to content

Commit 67911f3

Browse files
[PATCH] VLLM CVE Patch (#5031)
* build vllm * build vllm * revert toml * revert buildspec * moved script to docker file * fixed && * revert toml * build vllm * build vllm without autopatch * revert toml
1 parent 5807c97 commit 67911f3

File tree

1 file changed

+13
-3
lines changed

1 file changed

+13
-3
lines changed

vllm/x86_64/gpu/Dockerfile

Lines changed: 13 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -42,14 +42,24 @@ RUN chmod +x /usr/local/bin/deep_learning_container.py && \
4242
${HOME_DIR}/oss_compliance/generate_oss_compliance.sh ${HOME_DIR} ${PYTHON} && \
4343
# create symlink for python
4444
ln -s /usr/bin/python3 /usr/bin/python && \
45-
# remove cuobjdump and nvdisasm
46-
rm -rf /usr/local/cuda/bin/cuobjdump* && \
47-
rm -rf /usr/local/cuda/bin/nvdisasm* && \
4845
# clean up
4946
rm -rf ${HOME_DIR}/oss_compliance* && \
5047
rm -rf /tmp/tmp* && \
5148
rm -rf /tmp/uv* && \
5249
rm -rf /var/lib/apt/lists/* && \
5350
rm -rf /root/.cache | true
5451

52+
RUN mkdir -p /tmp/nvjpeg \
53+
&& cd /tmp/nvjpeg \
54+
&& wget https://developer.download.nvidia.com/compute/cuda/redist/libnvjpeg/linux-x86_64/libnvjpeg-linux-x86_64-12.4.0.76-archive.tar.xz \
55+
&& tar -xvf libnvjpeg-linux-x86_64-12.4.0.76-archive.tar.xz \
56+
&& rm -rf /usr/local/cuda/targets/x86_64-linux/lib/libnvjpeg* \
57+
&& rm -rf /usr/local/cuda/targets/x86_64-linux/include/nvjpeg.h \
58+
&& cp libnvjpeg-linux-x86_64-12.4.0.76-archive/lib/libnvjpeg* /usr/local/cuda/targets/x86_64-linux/lib/ \
59+
&& cp libnvjpeg-linux-x86_64-12.4.0.76-archive/include/* /usr/local/cuda/targets/x86_64-linux/include/ \
60+
&& rm -rf /tmp/nvjpeg \
61+
# remove cuobjdump and nvdisasm
62+
&& rm -rf /usr/local/cuda/bin/cuobjdump* \
63+
&& rm -rf /usr/local/cuda/bin/nvdisasm*
64+
5565
ENTRYPOINT ["/usr/local/bin/dockerd_entrypoint.sh"]

0 commit comments

Comments
 (0)