diff --git a/requirements-tensorrt.txt b/requirements-tensorrt.txt index 90517babd..d6027d0b4 100644 --- a/requirements-tensorrt.txt +++ b/requirements-tensorrt.txt @@ -1,7 +1,7 @@ # NVidia TensorRT Support (amd64 only) nvidia-pyindex; platform_machine == 'x86_64' nvidia-tensorrt == 8.4.1.5; platform_machine == 'x86_64' -cuda-python == 11.7; platform_machine == 'x86_64' +cuda-python == 12.0.0; platform_machine == 'x86_64' cython == 0.29.*; platform_machine == 'x86_64' nvidia-cuda-runtime-cu11 == 11.7.*; platform_machine == 'x86_64' nvidia-cublas-cu11 == 11.11.*; platform_machine == 'x86_64'