From 123e56467ecdf24f4b23623f09ee8b509b6a41af Mon Sep 17 00:00:00 2001 From: Nicolas Mowen Date: Sun, 8 Jun 2025 16:10:27 -0600 Subject: [PATCH] Expand packages for tensorflow model training --- docker/tensorrt/Dockerfile.amd64 | 2 +- docker/tensorrt/requirements-amd64.txt | 22 +++++++++++++++------- 2 files changed, 16 insertions(+), 8 deletions(-) diff --git a/docker/tensorrt/Dockerfile.amd64 b/docker/tensorrt/Dockerfile.amd64 index 8e0d756f6..906e113a8 100644 --- a/docker/tensorrt/Dockerfile.amd64 +++ b/docker/tensorrt/Dockerfile.amd64 @@ -18,7 +18,7 @@ FROM deps AS frigate-tensorrt ARG PIP_BREAK_SYSTEM_PACKAGES RUN --mount=type=bind,from=trt-wheels,source=/trt-wheels,target=/deps/trt-wheels \ - pip3 uninstall -y onnxruntime-openvino \ + pip3 uninstall -y onnxruntime-openvino tensorflow-cpu \ && pip3 install -U /deps/trt-wheels/*.whl COPY --from=rootfs / / diff --git a/docker/tensorrt/requirements-amd64.txt b/docker/tensorrt/requirements-amd64.txt index 615478549..91700b442 100644 --- a/docker/tensorrt/requirements-amd64.txt +++ b/docker/tensorrt/requirements-amd64.txt @@ -1,12 +1,20 @@ # NVidia TensorRT Support (amd64 only) --extra-index-url 'https://pypi.nvidia.com' -cython == 3.0.*; platform_machine == 'x86_64' -nvidia-cuda-runtime-cu12 == 12.1.*; platform_machine == 'x86_64' -nvidia-cublas-cu12==12.6.4.*; platform_machine == 'x86_64' -nvidia-cudnn-cu12 == 9.5.0.*; platform_machine == 'x86_64' -nvidia-cufft-cu12==11.*; platform_machine == 'x86_64' -nvidia-curand-cu12==10.3.9.*; platform_machine == 'x86_64' -nvidia-cuda-nvrtc-cu12==12.6.*; platform_machine == 'x86_64' +cython==3.0.*; platform_machine == 'x86_64' +nvidia_cuda_cupti_cu12==12.5.82; platform_machine == 'x86_64' +nvidia-cuda-runtime-cu12==12.1.*; platform_machine == 'x86_64' +nvidia-cublas-cu12==12.5.3.*; platform_machine == 'x86_64' +nvidia-cudnn-cu12==9.3.0.*; platform_machine == 'x86_64' +nvidia-cufft-cu12==11.2.3.*; platform_machine == 'x86_64' +nvidia-curand-cu12==10.3.6.*; platform_machine == 'x86_64' +nvidia_cuda_nvcc_cu12==12.5.82; platform_machine == 'x86_64' +nvidia-cuda-nvrtc-cu12==12.5.82; platform_machine == 'x86_64' +nvidia_cuda_runtime_cu12==12.5.82; platform_machine == 'x86_64' +nvidia_cusolver_cu12==11.6.3.*; platform_machine == 'x86_64' +nvidia_cusparse_cu12==12.5.1.*; platform_machine == 'x86_64' +nvidia_nccl_cu12==2.23.4; platform_machine == 'x86_64' +nvidia_nvjitlink_cu12==12.5.82; platform_machine == 'x86_64' +tensorflow==2.19.*; platform_machine == 'x86_64' onnx==1.16.*; platform_machine == 'x86_64' onnxruntime-gpu==1.22.*; platform_machine == 'x86_64' protobuf==3.20.3; platform_machine == 'x86_64'