From 71d8f447a004949c31c4d1abadf0351d292a2938 Mon Sep 17 00:00:00 2001 From: Emma Date: Tue, 10 Oct 2023 11:38:42 +0800 Subject: [PATCH] Update for onnxruntime (#1075) --- docker/dockerfile.merlin | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/docker/dockerfile.merlin b/docker/dockerfile.merlin index 77e08edbe..d4d95c4da 100644 --- a/docker/dockerfile.merlin +++ b/docker/dockerfile.merlin @@ -94,7 +94,7 @@ RUN ln -s /usr/bin/python3 /usr/bin/python # A fix has already been merged but not yet released: # https://gitlab.kitware.com/cmake/cmake/-/merge_requests/7859 # 2023-02-22: pynvml==11.5.0 is currently incompatible with our version of dask/distributed - +# 2023-10-06: onnxruntime==1.15.1 the latest version changed api which is not compatible with hugectr RUN pip install --no-cache-dir --upgrade pip; pip install --no-cache-dir "cmake<3.25.0" ninja scikit-build pandas==1.5.2 \ fastrlock nvidia-pyindex pybind11 pytest \ transformers==4.27.1 tensorflow-metadata betterproto \ @@ -106,7 +106,7 @@ RUN pip install --no-cache-dir --upgrade pip; pip install --no-cache-dir "cmake< numba "cuda-python>=11.5,<12.0" fsspec==2022.5.0 llvmlite \ pynvml==11.4.1 RUN pip install --no-cache-dir treelite==2.4.0 treelite_runtime==2.4.0 -RUN pip install --no-cache-dir numpy==1.22.4 protobuf==3.20.3 onnx onnxruntime pycuda +RUN pip install --no-cache-dir numpy==1.22.4 protobuf==3.20.3 onnx onnxruntime==1.15.1 pycuda RUN pip install --no-cache-dir dask==${DASK_VER} distributed==${DASK_VER} dask[dataframe]==${DASK_VER} RUN pip install --no-cache-dir onnx_graphsurgeon --index-url https://pypi.ngc.nvidia.com