Split cuda install commands in Dockerfile

Dockerfile

@@ -7,12 +7,14 @@ LABEL org.opencontainers.image.source = "https://github.com/aqlaboratory/openfol
 LABEL org.opencontainers.image.licenses = "Apache License 2.0"
 LABEL org.opencontainers.image.base.name="docker.io/nvidia/cuda:12.4.1-devel-ubuntu22.04"
 
-RUN apt-get update && apt-get install -y wget libxml2 cuda-minimal-build-12-4 libcusparse-dev-12-4 libcublas-dev-12-4 libcusolver-dev-12-4 git
+RUN apt-get update && apt-get install -y wget
 
 RUN apt-key del 7fa2af80
 RUN wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.0-1_all.deb
 RUN dpkg -i cuda-keyring_1.0-1_all.deb
 
+RUN apt-get install -y libxml2 cuda-minimal-build-12-1 libcusparse-dev-12-1 libcublas-dev-12-1 libcusolver-dev-12-1 git
+
 RUN wget -P /tmp \
     "https://github.com/conda-forge/miniforge/releases/download/23.3.1-1/Miniforge3-Linux-x86_64.sh" \
     && bash /tmp/Miniforge3-Linux-x86_64.sh -b -p /opt/conda \