Skip to content

Commit

Permalink
fix solver for cuml/cudf
Browse files Browse the repository at this point in the history
  • Loading branch information
djherbis committed Oct 4, 2024
1 parent 6b5b81b commit ec53f2e
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion Dockerfile.tmpl
Original file line number Diff line number Diff line change
Expand Up @@ -106,7 +106,6 @@ 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 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 }}
Expand Down

0 comments on commit ec53f2e

Please sign in to comment.