diff --git a/Dockerfile b/Dockerfile index d7074bf53..660cb5b25 100644 --- a/Dockerfile +++ b/Dockerfile @@ -283,7 +283,6 @@ COPY docker/support/tensorrt_detector/rootfs/ / RUN --mount=type=bind,from=trt-wheels,source=/trt-wheels,target=/deps/trt-wheels \ pip3 install -U /deps/trt-wheels/*.whl && \ - ln -s libnvrtc.so.12 /usr/local/lib/python3.9/dist-packages/nvidia/cuda_nvrtc/lib/libnvrtc.so && \ ldconfig # Dev Container w/ TRT diff --git a/requirements-tensorrt.txt b/requirements-tensorrt.txt index e6a337a86..214202e43 100644 --- a/requirements-tensorrt.txt +++ b/requirements-tensorrt.txt @@ -2,11 +2,11 @@ --extra-index-url 'https://pypi.nvidia.com' numpy < 1.24; platform_machine == 'x86_64' tensorrt == 8.5.3.*; platform_machine == 'x86_64' -cuda-python == 12.1; platform_machine == 'x86_64' +cuda-python == 11.8; platform_machine == 'x86_64' cython == 0.29.*; platform_machine == 'x86_64' nvidia-cuda-runtime-cu12 == 12.1.*; platform_machine == 'x86_64' -nvidia-cublas-cu12 == 12.1.*; platform_machine == 'x86_64' -nvidia-cudnn-cu12 == 8.8.*; platform_machine == 'x86_64' -nvidia-cuda-nvrtc-cu12 == 12.1.*; platform_machine == 'x86_64' +nvidia-cuda-runtime-cu11 == 11.8.*; platform_machine == 'x86_64' +nvidia-cublas-cu11 == 11.11.3.6; platform_machine == 'x86_64' +nvidia-cudnn-cu11 == 8.6.0.*; platform_machine == 'x86_64' onnx==1.14.0; platform_machine == 'x86_64' protobuf==3.20.3; platform_machine == 'x86_64' \ No newline at end of file