ARG BASE

ARG PYTHON_PACKAGE_MANAGER=conda

FROM ${BASE} as pip-base

RUN apt update -y \

&& DEBIAN_FRONTEND=noninteractive apt install -y --no-install-recommends \

# faiss dependencies

libblas-dev \

liblapack-dev \

&& rm -rf /tmp/* /var/tmp/* /var/cache/apt/* /var/lib/apt/lists/*;

ENV DEFAULT_VIRTUAL_ENV=rapids

ENV RAPIDS_LIBUCX_PREFER_SYSTEM_LIBRARY=true

FROM ${BASE} as conda-base

ENV DEFAULT_CONDA_ENV=rapids

FROM ${PYTHON_PACKAGE_MANAGER}-base

ARG TARGETARCH

ARG CUDA

ENV CUDAARCHS="RAPIDS"

ENV CUDA_VERSION="${CUDA_VERSION:-${CUDA}}"

ARG PYTHON_PACKAGE_MANAGER

ENV PYTHON_PACKAGE_MANAGER="${PYTHON_PACKAGE_MANAGER}"

ENV PYTHONSAFEPATH="1"

ENV PYTHONUNBUFFERED="1"

ENV PYTHONDONTWRITEBYTECODE="1"

ENV HISTFILE="/home/coder/.cache/._bash_history"

ENV AWS_IDP_URL="https://token.rapids.nvidia.com"

ENV AWS_ROLE_ARN="arn:aws:iam::279114543810:role/rapids-token-sccache-devs"

ENV SCCACHE_REGION="us-east-2"

ENV SCCACHE_BUCKET="rapids-sccache-devs"

ENV SCCACHE_S3_USE_PREPROCESSOR_CACHE_MODE=true

ENV SCCACHE_IDLE_TIMEOUT=0

ENV DEVCONTAINER_UTILS_ENABLE_SCCACHE_DIST=1

ENV SCCACHE_DIST_FALLBACK_TO_LOCAL_COMPILE=true

ENV SCCACHE_DIST_MAX_RETRIES=4

ENV SCCACHE_DIST_REQUEST_TIMEOUT=7140

ENV SCCACHE_DIST_URL="https://${TARGETARCH}.linux.sccache.rapids.nvidia.com"

ENV INFER_NUM_DEVICE_ARCHITECTURES=1

ENV MAX_DEVICE_OBJ_TO_COMPILE_IN_PARALLEL=20