From 2065d648b4583f715226a31b8492b2d4c8ac6471 Mon Sep 17 00:00:00 2001 From: Nate Meyer Date: Sun, 2 Jul 2023 00:28:34 -0400 Subject: [PATCH] Add numpy requirement prior to removal of np.bool --- requirements-tensorrt.txt | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) 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'