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

Add PIP_EXTRA_INDEX_URL and gh tool to build images #29

Merged
merged 10 commits into from
Jul 19, 2023
14 changes: 14 additions & 0 deletions ci-wheel/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@ FROM ${BASE_IMAGE}

ARG CUDA_VER
ARG LINUX_VER
ARG CPU_ARCH
ARG REAL_ARCH
ARG PYTHON_VER=3.9
ARG MANYLINUX_VER
Expand All @@ -18,6 +19,9 @@ ARG DEBIAN_FRONTEND=noninteractive
ENV RAPIDS_CUDA_VERSION="${CUDA_VER}"
ENV RAPIDS_PY_VERSION="${PYTHON_VER}"

# RAPIDS pip index
ENV PIP_EXTRA_INDEX_URL="https://pypi.k8s.rapids.ai/simple"

ENV PYENV_ROOT="/pyenv"
ENV PATH="/pyenv/bin:/pyenv/shims:$PATH"

Expand Down Expand Up @@ -48,6 +52,16 @@ RUN case "${LINUX_VER}" in \
;; \
esac

# Download and install GH CLI tool v2.32.0
ARG GH_VERSION=2.32.0
RUN <<EOF
set -e
wget https://github.com/cli/cli/releases/download/v${GH_VERSION}/gh_${GH_VERSION}_linux_${CPU_ARCH}.tar.gz
tar -xf gh_*.tar.gz
mv gh_*/bin/gh /usr/local/bin
rm -rf gh_*
EOF

# Install sccache
ARG SCCACHE_VERSION=0.5.0

Expand Down