diff --git a/requirements-tensorrt.txt b/requirements-tensorrt.txt index a6cfd9bb8..e6a337a86 100644 --- a/requirements-tensorrt.txt +++ b/requirements-tensorrt.txt @@ -1,8 +1,7 @@ # NVidia TensorRT Support (amd64 only) --extra-index-url 'https://pypi.nvidia.com' +numpy < 1.24; platform_machine == 'x86_64' tensorrt == 8.5.3.*; platform_machine == 'x86_64' -# tensorrt-libs == 8.5.3.*; platform_machine == 'x86_64' -# tensorrt-bindings == 8.5.3.*; platform_machine == 'x86_64' cuda-python == 12.1; platform_machine == 'x86_64' cython == 0.29.*; platform_machine == 'x86_64' nvidia-cuda-runtime-cu12 == 12.1.*; platform_machine == 'x86_64'