From 196fbdd818803c4956e1cbdf9543293bc94a17f8 Mon Sep 17 00:00:00 2001 From: Dante Gama Dessavre Date: Wed, 27 Sep 2023 17:10:52 -0500 Subject: [PATCH] FIX mamba clean command : --- raft-ann-bench/cpu/Dockerfile | 3 ++- raft-ann-bench/gpu/Dockerfile | 6 +++--- 2 files changed, 5 insertions(+), 4 deletions(-) diff --git a/raft-ann-bench/cpu/Dockerfile b/raft-ann-bench/cpu/Dockerfile index 6e5b86e8..e39b0be7 100644 --- a/raft-ann-bench/cpu/Dockerfile +++ b/raft-ann-bench/cpu/Dockerfile @@ -24,7 +24,8 @@ WORKDIR /home/rapids COPY condarc /opt/conda/.condarc -RUN mamba install -y -n base \ +RUN mamba update --all -y -n base \ + &&mamba install -y -n base \ "raft-ann-bench-cpu=${RAPIDS_VER}.*" \ "python=${PYTHON_VER}" \ && mamba update --all -y -n base \ diff --git a/raft-ann-bench/gpu/Dockerfile b/raft-ann-bench/gpu/Dockerfile index 843f04c0..2d70599c 100644 --- a/raft-ann-bench/gpu/Dockerfile +++ b/raft-ann-bench/gpu/Dockerfile @@ -4,7 +4,7 @@ ARG CUDA_VER=12.0.1 ARG PYTHON_VER=3.10 ARG LINUX_VER=ubuntu22.04 -ARG RAPIDS_VER=23.10 +ARG RAPIDS_VER=23.12 FROM rapidsai/mambaforge-cuda:cuda${CUDA_VER}-base-${LINUX_VER}-py${PYTHON_VER} as raft-ann-bench ARG CUDA_VER @@ -19,11 +19,11 @@ WORKDIR /home/rapids COPY condarc /opt/conda/.condarc -RUN mamba install -y -n base \ +RUN mamba update --all -y -n base \ + && mamba install -y -n base \ "raft-ann-bench=${RAPIDS_VER}.*" \ "cuda-version=${CUDA_VER%.*}.*" \ "rmm=${RAPIDS_VER}.*" \ - && mamba update --all -y -n base \ && conda clean -afy # we need to change the permissions of the conda folder so that all users can execute