From 3f4e03185165f508c1ef18a27b0c06cc732aecbd Mon Sep 17 00:00:00 2001 From: qqiao Date: Sat, 6 Jul 2024 01:50:38 +0000 Subject: [PATCH] Add numpy because it is needed by hps_torch_plugin --- docker/dockerfile.merlin.ctr | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docker/dockerfile.merlin.ctr b/docker/dockerfile.merlin.ctr index d797702fc..53275a008 100644 --- a/docker/dockerfile.merlin.ctr +++ b/docker/dockerfile.merlin.ctr @@ -82,7 +82,7 @@ RUN pip install --no-cache-dir --upgrade pip; pip install --no-cache-dir "cmake< nvidia-pyindex pybind11 pytest \ "scikit-learn<1.2" \ tritonclient[all] -RUN pip install --no-cache-dir onnx onnxruntime pycuda +RUN pip install --no-cache-dir numpy==1.22.4 onnx onnxruntime pycuda RUN pip install --no-cache-dir onnx_graphsurgeon --index-url https://pypi.ngc.nvidia.com # Triton Server