Skip to content

Migrate to micromamba #1431

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.

Sign up for GitHub

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

Merged
merged 6 commits into from
Oct 4, 2024
Merged

Migrate to micromamba #1431

Changes from 1 commit
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
f1f0135
Migrate to micromamba
djherbis Oct 3, 2024
fdbc49b
fix paths and tests
djherbis Oct 3, 2024
fc5c14b
trying to fix gpu image
djherbis Oct 4, 2024
c02f4c1
allow flexible channels
djherbis Oct 4, 2024
6b5b81b
fix cudf/cuml version pins with micromamba
djherbis Oct 4, 2024
ec53f2e
fix solver for cuml/cudf
djherbis Oct 4, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
allow flexible channels
  • Loading branch information
@djherbis
djherbis committed Oct 4, 2024
commit c02f4c129a342e08f40198213a2fe5096367bdee
6 changes: 3 additions & 3 deletions Dockerfile.tmpl
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -88,10 +88,10 @@ RUN curl -L "https://micro.mamba.pm/install.sh" -o /tmp/micromamba-install.sh \
&& rm /tmp/micromamba-install.sh \
&& mv ~/.local/bin/micromamba /usr/bin/micromamba \
&& (!(which conda) || cp /usr/bin/micromamba $(which conda)) \
&& micromamba config append channels conda-forge \
&& micromamba config append channels nvidia \
&& micromamba config append channels rapidsai \
&& micromamba config set channel_priority strict \
&& micromamba config append channels conda-forge \
&& micromamba config set channel_priority flexible \
&& python -m nb_conda_kernels.install --disable

# Install conda packages not available on pip.
Expand All @@ -106,7 +106,7 @@ 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 -y spacy cudf>=24.4 cuml>=24.4 cupy cuda-version=$CUDA_MAJOR_VERSION.$CUDA_MINOR_VERSION && \
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 && \
Expand Down