Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

[v1.x] Fix nightly cd cu102 #19940

Merged
merged 7 commits into from
Feb 21, 2021
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
14 changes: 9 additions & 5 deletions cd/python/docker/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -23,11 +23,15 @@
ARG BASE_IMAGE
FROM ${BASE_IMAGE}

ARG PYTHON_CMD=python3
RUN apt-get update && \
apt-get install -y wget ${PYTHON_CMD}-dev gcc && \
wget https://bootstrap.pypa.io/get-pip.py && \
${PYTHON_CMD} get-pip.py
RUN apt-get update || true
RUN apt-get install -y software-properties-common
RUN add-apt-repository -y ppa:deadsnakes/ppa
RUN apt-get update || true
RUN apt-get install -y python3.7-dev python3.7-distutils virtualenv wget
RUN ln -sf /usr/bin/python3.7 /usr/local/bin/python3

RUN wget -nv https://bootstrap.pypa.io/get-pip.py
RUN python3 get-pip.py

ARG MXNET_COMMIT_ID
ENV MXNET_COMMIT_ID=${MXNET_COMMIT_ID}
Expand Down
2 changes: 1 addition & 1 deletion tools/setup_gpu_build_tools.sh
Original file line number Diff line number Diff line change
Expand Up @@ -42,7 +42,7 @@ elif [[ $VARIANT == cu102* ]]; then
CUDA_VERSION='10.2.89-1'
CUDA_PATCH_VERSION='10.2.2.89-1'
LIBCUDA_VERSION='440.33.01-0ubuntu1'
LIBCUDNN_VERSION='7.6.5.32-1+cuda10.2'
LIBCUDNN_VERSION='8.0.4.30-1+cuda10.2'
LIBNCCL_VERSION='2.5.6-1+cuda10.2'
elif [[ $VARIANT == cu101* ]]; then
CUDA_VERSION='10.1.105-1'
Expand Down