From e4ea35e626a2354747a1dfee67cb4fba4f95e056 Mon Sep 17 00:00:00 2001 From: Nicolas Mowen Date: Sun, 15 Sep 2024 19:30:30 -0600 Subject: [PATCH] Add onnxruntime nvidia providers (#13756) --- docker/tensorrt/requirements-amd64.txt | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/docker/tensorrt/requirements-amd64.txt b/docker/tensorrt/requirements-amd64.txt index 214202e43..6cbfacd40 100644 --- a/docker/tensorrt/requirements-amd64.txt +++ b/docker/tensorrt/requirements-amd64.txt @@ -9,4 +9,5 @@ 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 +onnxruntime-gpu==1.18.0; platform_machine == 'x86_64' +protobuf==3.20.3; platform_machine == 'x86_64'