Fix docker build failure.
This commit is contained in:
@ -7,7 +7,7 @@ WORKDIR /
|
|||||||
|
|
||||||
RUN apt-get update && apt-get install -y --no-install-recommends feh
|
RUN apt-get update && apt-get install -y --no-install-recommends feh
|
||||||
RUN apt-get update && apt-get install -y --no-install-recommends python3-opencv
|
RUN apt-get update && apt-get install -y --no-install-recommends python3-opencv
|
||||||
RUN pip uninstall -y opencv-python && pip install --no-cache "opencv-python<4.3"
|
RUN pip uninstall -y opencv-python && pip install --no-cache-dir "opencv-python<4.3"
|
||||||
|
|
||||||
# Show matplotlib images
|
# Show matplotlib images
|
||||||
RUN apt-get update && apt-get install -y --no-install-recommends python3-tk
|
RUN apt-get update && apt-get install -y --no-install-recommends python3-tk
|
||||||
@ -23,7 +23,12 @@ RUN apt-get update && apt-get install -y --no-install-recommends libopencv-dev
|
|||||||
# Install ONNX runtime
|
# Install ONNX runtime
|
||||||
# See: https://github.com/microsoft/onnxruntime/blob/main/dockerfiles/Dockerfile.tensorrt
|
# See: https://github.com/microsoft/onnxruntime/blob/main/dockerfiles/Dockerfile.tensorrt
|
||||||
RUN pip3 uninstall -y onnxruntime-gpu
|
RUN pip3 uninstall -y onnxruntime-gpu
|
||||||
|
RUN pip3 install --no-cache-dir psutil
|
||||||
RUN git clone --recursive --depth=1 --branch=v1.20.1 https://github.com/Microsoft/onnxruntime.git
|
RUN git clone --recursive --depth=1 --branch=v1.20.1 https://github.com/Microsoft/onnxruntime.git
|
||||||
|
# Next line fixes: https://github.com/microsoft/onnxruntime/issues/24861
|
||||||
|
RUN cat /onnxruntime/cmake/deps.txt && \
|
||||||
|
sed -i 's/;be8be39fdbc6e60e94fa7870b280707069b5b81a/;32b145f525a8308d7ab1c09388b2e288312d8eba/g' /onnxruntime/cmake/deps.txt && \
|
||||||
|
cat /onnxruntime/cmake/deps.txt
|
||||||
ENV PATH=/usr/local/nvidia/bin:/usr/local/cuda/bin:/cmake-3.30.1-linux-x86_64/bin:${PATH}
|
ENV PATH=/usr/local/nvidia/bin:/usr/local/cuda/bin:/cmake-3.30.1-linux-x86_64/bin:${PATH}
|
||||||
ARG CMAKE_CUDA_ARCHITECTURES=75;80;90
|
ARG CMAKE_CUDA_ARCHITECTURES=75;80;90
|
||||||
ENV TRT_VERSION=10.5.0.18
|
ENV TRT_VERSION=10.5.0.18
|
||||||
|
|||||||
Reference in New Issue
Block a user