Skip to content

Commit

Permalink
Remove other debug info
Browse files Browse the repository at this point in the history
  • Loading branch information
EmmaQiaoCh committed Dec 14, 2023
1 parent 89e5146 commit 87216f7
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 3 deletions.
4 changes: 2 additions & 2 deletions docker/dockerfile.merlin
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down Expand Up @@ -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
Expand Down
1 change: 0 additions & 1 deletion docker/dockerfile.tf
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 87216f7

Please sign in to comment.