File tree Expand file tree Collapse file tree 1 file changed +4
-3
lines changed Expand file tree Collapse file tree 1 file changed +4
-3
lines changed Original file line number Diff line number Diff line change 33FROM nvcr.io/nvidia/l4t-jetpack:r36.4.0 AS builder
44
55ARG DEBIAN_FRONTEND=noninteractive
6- ARG CMAKE_VERSION=3.28.6
6+ ARG CMAKE_VERSION=3.31.10
77ARG PYTORCH_VERSION=2.8.0
88ARG TORCHVISION_VERSION=0.23.0
99ARG OPENCV_VERSION=4.10.0
@@ -140,10 +140,11 @@ RUN uv pip install --system --break-system-packages --index-strategy unsafe-best
140140 packaging \
141141 && rm -rf ~/.cache/uv
142142
143- # Build onnxruntime with numpy 2.x
143+ # Build onnxruntime-gpu from source with TensorRT support
144144WORKDIR /build/onnxruntime
145145RUN git clone --recursive --branch v${ONNXRUNTIME_VERSION} https://github.com/microsoft/onnxruntime.git && \
146146 cd onnxruntime && \
147+ sed -i 's/be8be39fdbc6e60e94fa7870b280707069b5b81a/32b145f525a8308d7ab1c09388b2e288312d8eba/g' cmake/deps.txt && \
147148 ./build.sh \
148149 --config Release \
149150 --build_dir build/cuda12 \
@@ -162,7 +163,7 @@ RUN git clone --recursive --branch v${ONNXRUNTIME_VERSION} https://github.com/mi
162163 --allow_running_as_root \
163164 --cmake_extra_defines CMAKE_CUDA_ARCHITECTURES="87" \
164165 --cmake_extra_defines onnxruntime_BUILD_UNIT_TESTS=OFF && \
165- python3 -m pip install build/cuda12/Release/dist/onnxruntime_gpu-*.whl
166+ python3 -m pip install --break-system-packages build/cuda12/Release/dist/onnxruntime_gpu-*.whl
166167
167168# Build inference packages
168169WORKDIR /build/inference
You can’t perform that action at this time.
0 commit comments