Add symlink to libnvrtc

This commit is contained in:
Nate Meyer 2023-01-14 11:37:24 -05:00
parent 0baae653dd
commit 9c641eca41

View File

@ -270,6 +270,7 @@ COPY --from=rootfs / /
FROM frigate AS frigate-tensorrt FROM frigate AS frigate-tensorrt
RUN --mount=type=bind,from=trt-wheels,source=/trt-wheels,target=/deps/trt-wheels \ RUN --mount=type=bind,from=trt-wheels,source=/trt-wheels,target=/deps/trt-wheels \
pip3 install -U /deps/trt-wheels/*.whl && \ pip3 install -U /deps/trt-wheels/*.whl && \
ln -s libnvrtc.so.11.2 /usr/local/lib/python3.9/dist-packages/nvidia/cuda_nvrtc/lib/libnvrtc.so && \
ldconfig ldconfig
# Dev Container w/ TRT # Dev Container w/ TRT