File tree Expand file tree Collapse file tree 1 file changed +13
-3
lines changed
Expand file tree Collapse file tree 1 file changed +13
-3
lines changed Original file line number Diff line number Diff 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+
5565ENTRYPOINT ["/usr/local/bin/dockerd_entrypoint.sh" ]
You can’t perform that action at this time.
0 commit comments