Revert numpy upgrade in -tensorrt image (#7348)

blakeblackshear · Aug 5, 2023 · 318ae28 · 318ae28
1 parent 24365fb
commit 318ae28
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docker/tensorrt/requirements-amd64.txt b/docker/tensorrt/requirements-amd64.txt
@@ -1,6 +1,6 @@
 # NVidia TensorRT Support (amd64 only)
 --extra-index-url 'https://pypi.nvidia.com'
-numpy <= 1.25; platform_machine == 'x86_64'
+numpy < 1.24; platform_machine == 'x86_64'
 tensorrt == 8.5.3.*; platform_machine == 'x86_64'
 cuda-python == 11.8; platform_machine == 'x86_64'
 cython == 0.29.*; platform_machine == 'x86_64'