diff --git a/docker/tensorrt/requirements-amd64.txt b/docker/tensorrt/requirements-amd64.txt index caa65cdb6..3cf0fb2e3 100644 --- a/docker/tensorrt/requirements-amd64.txt +++ b/docker/tensorrt/requirements-amd64.txt @@ -13,6 +13,6 @@ nvidia-cusolver-cu12==11.7.5.*; platform_machine == 'x86_64' nvidia-cusparse-cu12==12.5.10.*; platform_machine == 'x86_64' nvidia-nccl-cu12==2.29.7; platform_machine == 'x86_64' nvidia-nvjitlink-cu12==12.9.86; platform_machine == 'x86_64' -onnx==1.16.*; platform_machine == 'x86_64' +onnx==1.21.*; platform_machine == 'x86_64' onnxruntime-gpu==1.24.*; platform_machine == 'x86_64' protobuf==3.20.3; platform_machine == 'x86_64' diff --git a/docker/tensorrt/requirements-models-arm64.txt b/docker/tensorrt/requirements-models-arm64.txt index fe89b4754..6855dbd4d 100644 --- a/docker/tensorrt/requirements-models-arm64.txt +++ b/docker/tensorrt/requirements-models-arm64.txt @@ -1,2 +1,2 @@ -onnx == 1.14.0; platform_machine == 'aarch64' +onnx == 1.21.0rc1; platform_machine == 'aarch64' protobuf == 3.20.3; platform_machine == 'aarch64'