Skip to content

Commit c4ee63d

Browse files
committed
rm micromamba since its not actually applying to the env
1 parent 0d6c8ed commit c4ee63d

File tree

1 file changed

+0
-17
lines changed

1 file changed

+0
-17
lines changed

Dockerfile.tmpl

Lines changed: 0 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -26,15 +26,6 @@ ADD patches/template_conf.json /opt/kaggle/conf.json
2626
# /opt/conda/lib/python3.10/site-packages
2727
ARG PACKAGE_PATH=/usr/local/lib/python3.10/dist-packages
2828

29-
# Install micromamba
30-
RUN curl -L micro.mamba.pm/install.sh | sh
31-
32-
# Install conda packages not available on pip.
33-
# When using pip in a conda environment, conda commands should be ran first and then
34-
# the remaining pip commands: https://www.anaconda.com/using-pip-in-a-conda-environment/
35-
RUN micromamba install -y mkl imagemagick pyproj && \
36-
/tmp/clean-layer.sh
37-
3829
# Install GPU-specific non-pip packages.
3930
{{ if eq .Accelerator "gpu" }}
4031
ARG CUDA_MAJOR_VERSION \
@@ -45,14 +36,6 @@ ENV CUDA_MAJOR_VERSION=${CUDA_MAJOR_VERSION} \
4536
# Make sure we are on the right version of CUDA
4637
RUN update-alternatives --set cuda /usr/local/cuda-$CUDA_MAJOR_VERSION.$CUDA_MINOR_VERSION
4738

48-
RUN micromamba config append channels nvidia \
49-
&& micromamba config append channels rapidsai \
50-
&& micromamba config append channels conda-forge \
51-
&& micromamba config set channel_priority flexible
52-
53-
RUN micromamba install -vvvy "cudf>=24.4" "cuml>=24.4" cupy cuda-version=$CUDA_MAJOR_VERSION.$CUDA_MINOR_VERSION && \
54-
/tmp/clean-layer.sh
55-
5639
# Remove CUDA_VERSION from non-GPU image.
5740
{{ else }}
5841
ENV CUDA_VERSION=""

0 commit comments

Comments
 (0)