fix cudf/cuml version pins with micromamba

Kaggle · Oct 4, 2024 · 6b5b81b · 6b5b81b
1 parent c02f4c1
commit 6b5b81b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/Dockerfile.tmpl b/Dockerfile.tmpl
@@ -106,7 +106,8 @@ RUN micromamba install -y mkl cartopy imagemagick pyproj "shapely<2" && \
 # b/341938540: unistall grpc-cpp to allow >=v24.4 cudf and cuml to be installed.
 {{ if eq .Accelerator "gpu" }}
 RUN pip uninstall -y pyarrow && \
-    micromamba install -vvvy spacy cudf>=24.4 cuml>=24.4 cupy cuda-version=$CUDA_MAJOR_VERSION.$CUDA_MINOR_VERSION && \
+    micromamba remove -y --force grpc-cpp && \
+    micromamba install -vvvy spacy "cudf>=24.4" "cuml>=24.4" cupy cuda-version=$CUDA_MAJOR_VERSION.$CUDA_MINOR_VERSION && \
     /tmp/clean-layer.sh
 {{ else }}
 RUN pip install spacy && \