File tree Expand file tree Collapse file tree 1 file changed +0
-17
lines changed
Expand file tree Collapse file tree 1 file changed +0
-17
lines changed Original file line number Diff line number Diff line change @@ -26,15 +26,6 @@ ADD patches/template_conf.json /opt/kaggle/conf.json
2626# /opt/conda/lib/python3.10 /site-packages
2727ARG 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" }}
4031ARG 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
4637RUN 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 }}
5841ENV CUDA_VERSION= " "
You can’t perform that action at this time.
0 commit comments