From 00903af3d0343cc08fb0580137740793753afad1 Mon Sep 17 00:00:00 2001 From: Nicolas Mowen Date: Thu, 24 Oct 2024 07:15:15 -0600 Subject: [PATCH] Update nvidia deps --- docker/tensorrt/requirements-amd64.txt | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/docker/tensorrt/requirements-amd64.txt b/docker/tensorrt/requirements-amd64.txt index b5ad4fcbd..c73797516 100644 --- a/docker/tensorrt/requirements-amd64.txt +++ b/docker/tensorrt/requirements-amd64.txt @@ -9,6 +9,6 @@ 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' nvidia-cufft-cu11==10.*; platform_machine == 'x86_64' -onnx==1.14.0; platform_machine == 'x86_64' -onnxruntime-gpu==1.17.*; platform_machine == 'x86_64' +onnx==1.16.*; platform_machine == 'x86_64' +onnxruntime-gpu==1.19.*; platform_machine == 'x86_64' protobuf==3.20.3; platform_machine == 'x86_64'