From 87216f7e1cadc8df928a75a7c22e9833eb7c1f37 Mon Sep 17 00:00:00 2001 From: qqiao Date: Thu, 14 Dec 2023 03:19:40 -0800 Subject: [PATCH] Remove other debug info --- docker/dockerfile.merlin | 4 ++-- docker/dockerfile.tf | 1 - 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/docker/dockerfile.merlin b/docker/dockerfile.merlin index eb8bdbdfe..d4d95c4da 100644 --- a/docker/dockerfile.merlin +++ b/docker/dockerfile.merlin @@ -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.0 protobuf==3.20.3 onnx onnxruntime==1.15.1 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 @@ -299,7 +299,7 @@ COPY --chown=1000:1000 --from=dlfw /usr/local/lib/python${PYTHON_VERSION}/dist-p COPY --chown=1000:1000 --from=dlfw /usr/local/lib/python${PYTHON_VERSION}/dist-packages/numba-*.dist-info /usr/local/lib/python${PYTHON_VERSION}/dist-packages/numba.dist-info/ COPY --chown=1000:1000 --from=dlfw /usr/local/lib/python${PYTHON_VERSION}/dist-packages/cubinlinker-*.dist-info /usr/local/lib/python${PYTHON_VERSION}/dist-packages/cubinlinker.dist-info/ -RUN pip install --no-cache-dir jupyterlab notebook pydot testbook numpy==1.22.0 +RUN pip install --no-cache-dir jupyterlab notebook pydot testbook numpy==1.22.4 ENV JUPYTER_CONFIG_DIR=/tmp/.jupyter ENV JUPYTER_DATA_DIR=/tmp/.jupyter diff --git a/docker/dockerfile.tf b/docker/dockerfile.tf index 2bc13d082..7e4418cc0 100644 --- a/docker/dockerfile.tf +++ b/docker/dockerfile.tf @@ -59,7 +59,6 @@ RUN if [ "$HUGECTR_DEV_MODE" == "false" ]; then \ rm -rf ${HUGECTR_HOME}/lib/libgtest* ${HUGECTR_HOME}/lib/pkgconfig/gtest* ${HUGECTR_HOME}/include/gtest && \ git clone --branch ${HUGECTR_VER} --depth 1 --recurse-submodules --shallow-submodules https://${_CI_JOB_TOKEN}${_HUGECTR_REPO} /hugectr && \ pushd /hugectr && \ - git log -n 3 && \ rm -rf .git/modules && \ pip --no-cache-dir install ninja tf2onnx && \ # Install SOK