feat: isolate the CUDA install

Author: lsorber

Dockerfile

@@ -14,12 +14,12 @@ RUN CONDA_ARCH=$([ "$TARGETARCH" = "arm64" ] && echo "aarch64" || echo "x86_64")
     /bin/bash ~/miniconda.sh -b -p /opt/conda && \
     rm ~/miniconda.sh
 ENV PATH=/opt/conda/bin:$PATH
-# ENV LD_LIBRARY_PATH=/opt/conda/lib:$LD_LIBRARY_PATH
 
 # Install CUDA and cuDNN.
 ARG CUDA_VERSION=11.8
 ARG CUDNN_VERSION=8.8
 RUN conda install --name base --yes conda-libmamba-solver && \
     conda config --set solver libmamba && \
-    conda install --name base --channel conda-forge --yes cudatoolkit="$CUDA_VERSION" cudnn="$CUDNN_VERSION" && \
+    conda create --name cuda --no-default-packages --channel conda-forge --yes cudatoolkit="$CUDA_VERSION" cudnn="$CUDNN_VERSION" && \
     conda clean --all --force-pkgs-dirs --yes
+ENV LD_LIBRARY_PATH=/opt/conda/envs/cuda/lib:$LD_LIBRARY_PATH