FIX mamba clean command

:
rapidsai · Sep 27, 2023 · 196fbdd · 196fbdd
1 parent ddd717e
commit 196fbdd
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git 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
@@ -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