Update Dockerfile

mosaicml · May 17, 2024 · 4752172 · 4752172
1 parent 390f3e5
commit 4752172
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/docker/Dockerfile b/docker/Dockerfile
@@ -97,13 +97,14 @@ RUN if [ -n "$CUDA_VERSION" ]; then \
         apt-key del 7fa2af80 && \
         mkdir -p /tmp/cuda-keyring && \
         if [ "$TARGETARCH" = "amd64" ]; then \
+            echo $LINUX_DISTRO/$LINUX_DISTRO_VERSION; \
             wget -P /tmp/cuda-keyring https://developer.download.nvidia.com/compute/cuda/repos/${LINUX_DISTRO}${LINUX_DISTRO_VERSION}/x86_64/cuda-keyring_1.0-1_all.deb; \
         else \
             wget -P /tmp/cuda-keyring https://developer.download.nvidia.com/compute/cuda/repos/${LINUX_DISTRO}${LINUX_DISTRO_VERSION}/arm64/cuda-keyring_1.0-1_all.deb; \
         fi && \
         dpkg -i /tmp/cuda-keyring/cuda-keyring_1.0-1_all.deb && \
         rm -rf /tmp/cuda-keyring; \
-    fi/tmp/cuda-keyring ; \
+        /tmp/cuda-keyring ; \
     fi