Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Update condaforge/miniforge3 #660

Closed
wants to merge 10 commits into from
28 changes: 12 additions & 16 deletions raft-ann-bench/cpu/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -3,13 +3,21 @@
ARG PYTHON_VER=3.11
ARG RAPIDS_VER=24.08

FROM condaforge/mambaforge:23.3.1-0 as raft-ann-bench-cpu
FROM mambaorg/micromamba:1.5.8 AS raft-ann-bench-cpu
ARG RAPIDS_VER
ARG PYTHON_VER

COPY condarc /opt/conda/.condarc

RUN micromamba install -y -n base \
"python=${PYTHON_VER}" \
"raft-ann-bench-cpu=${RAPIDS_VER}.*" \
"conda" && \
micromamba clean --all --yes


SHELL ["/bin/bash", "-euo", "pipefail", "-c"]
USER root

# Create a data folder accessible by any user so mounted volumes under it can be accessible
# when the user passes their uid to docker run with -u $(id -u)
Expand All @@ -21,22 +29,10 @@ echo ". /opt/conda/etc/profile.d/conda.sh; conda activate base" >> /etc/bash.bas
EOF

# we need perl temporarily for the remaining benchmark perl scripts
RUN apt-get install perl -y

# Install python before updating environment, otherwise Python 3.9 image
# runs into a solver conflict with truststore 0.8.0. This avoids the environment installing
# packages incompatible with python version needed before python itself is pinned to the correct version.
RUN <<EOF
mamba install -y -n base "python=${PYTHON_VER}"
mamba update --all -y -n base
mamba install -y -n base \
"raft-ann-bench-cpu=${RAPIDS_VER}.*" \
"python=${PYTHON_VER}"
conda clean -afy
chmod -R 777 /opt/conda
EOF
RUN apt-get update -y && apt-get install perl-base -y

RUN useradd -rm -d /home/rapids -s /bin/bash -u 1001 rapids
RUN chown -R rapids /opt/conda
USER rapids
WORKDIR /data/benchmarks

Expand All @@ -47,7 +43,7 @@ CMD ["--dataset fashion-mnist-784-euclidean", "", "--algorithms hnswlib"]
ENTRYPOINT ["/bin/bash", "/data/scripts/run_benchmark.sh"]


FROM bench-base as raft-ann-bench-cpu-datasets
FROM bench-base AS raft-ann-bench-cpu-datasets

SHELL ["/bin/bash", "-euo", "pipefail", "-c"]

Expand Down
Loading