This commit is contained in:
Morgan Funtowicz 2024-07-23 08:11:36 +00:00
parent 4c657ca158
commit 3c39ab5ac8
1 changed files with 1 additions and 1 deletions

View File

@ -73,7 +73,7 @@ RUN cargo chef cook --release --recipe-path recipe.json
ARG CUDA_ARCH_LIST
ENV CMAKE_PREFIX_PATH="/usr/local/mpi:/usr/local/tensorrt:$CMAKE_PREFIX_PATH"
ENV LD_LIBRARY_PATH="/usr/local/mpi/lib:$LD_LIBRARY_PATH"
ENV PKG_CONFIG_PATH="/usr/local/mpi/lib/pkgconfig:$PKG_CONFIG_PATH"a
ENV PKG_CONFIG_PATH="/usr/local/mpi/lib/pkgconfig:$PKG_CONFIG_PATH"
COPY . .
COPY --from=trt-builder /usr/local/tensorrt /usr/local/tensorrt