From 3e2b91181ecac499836dfe63e4e57f274d91e88a Mon Sep 17 00:00:00 2001 From: Nicolas Mowen Date: Mon, 23 Mar 2026 07:10:25 -0600 Subject: [PATCH] Downgrade CUDA to support 50 series without errors --- docker/tensorrt/requirements-amd64.txt | 24 ++++++++++++------------ 1 file changed, 12 insertions(+), 12 deletions(-) diff --git a/docker/tensorrt/requirements-amd64.txt b/docker/tensorrt/requirements-amd64.txt index caa65cdb6..597680c00 100644 --- a/docker/tensorrt/requirements-amd64.txt +++ b/docker/tensorrt/requirements-amd64.txt @@ -1,18 +1,18 @@ # Nvidia ONNX Runtime GPU Support --extra-index-url 'https://pypi.nvidia.com' cython==3.0.*; platform_machine == 'x86_64' -nvidia-cuda-cupti-cu12==12.9.79; platform_machine == 'x86_64' -nvidia-cublas-cu12==12.9.1.*; platform_machine == 'x86_64' -nvidia-cudnn-cu12==9.19.0.*; platform_machine == 'x86_64' -nvidia-cufft-cu12==11.4.1.*; platform_machine == 'x86_64' -nvidia-curand-cu12==10.3.10.*; platform_machine == 'x86_64' -nvidia-cuda-nvcc-cu12==12.9.86; platform_machine == 'x86_64' -nvidia-cuda-nvrtc-cu12==12.9.86; platform_machine == 'x86_64' -nvidia-cuda-runtime-cu12==12.9.79; platform_machine == 'x86_64' -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' +nvidia-cuda-cupti-cu12==12.8.90; platform_machine == 'x86_64' +nvidia-cublas-cu12==12.8.4.1; platform_machine == 'x86_64' +nvidia-cudnn-cu12==9.8.0.87; platform_machine == 'x86_64' +nvidia-cufft-cu12==11.3.3.83; platform_machine == 'x86_64' +nvidia-curand-cu12==10.3.9.90; platform_machine == 'x86_64' +nvidia-cuda-nvcc-cu12==12.8.93; platform_machine == 'x86_64' +nvidia-cuda-nvrtc-cu12==12.8.93; platform_machine == 'x86_64' +nvidia-cuda-runtime-cu12==12.8.90; platform_machine == 'x86_64' +nvidia-cusolver-cu12==11.7.3.90; platform_machine == 'x86_64' +nvidia-cusparse-cu12==12.5.8.93; platform_machine == 'x86_64' +nvidia-nccl-cu12==2.26.2.post1; platform_machine == 'x86_64' +nvidia-nvjitlink-cu12==12.8.93; platform_machine == 'x86_64' onnx==1.16.*; platform_machine == 'x86_64' onnxruntime-gpu==1.24.*; platform_machine == 'x86_64' protobuf==3.20.3; platform_machine == 'x86_64'