move it up

mosaicml · Nov 16, 2023 · 20e5b8b · 20e5b8b
1 parent c12d352
commit 20e5b8b
Showing 1 changed file with 9 additions and 10 deletions.
diff --git a/docker/Dockerfile b/docker/Dockerfile
@@ -354,6 +354,15 @@ RUN pip install --no-cache-dir --upgrade \
         ipython${IPYTHON_VERSION} \
         urllib3${URLLIB3_VERSION}
 
+##################################################
+# Override NVIDIA mistaken env var for 11.8 images
+##################################################
+ARG NVIDIA_REQUIRE_CUDA_OVERRIDE
+ENV NVIDIA_REQUIRE_CUDA=${NVIDIA_REQUIRE_CUDA_OVERRIDE:-$NVIDIA_REQUIRE_CUDA}
+
+# Echo the new env var
+RUN echo "NVIDIA_REQUIRE_CUDA_OVERRIDE=${NVIDIA_REQUIRE_CUDA_OVERRIDE}"
+RUN echo "NVIDIA_REQUIRE_CUDA=${NVIDIA_REQUIRE_CUDA}"
 
 ################
 # Composer Image
@@ -370,13 +379,3 @@ ARG DEBIAN_FRONTEND=noninteractive
 ARG COMPOSER_INSTALL_COMMAND
 
 RUN pip install "${COMPOSER_INSTALL_COMMAND}"
-
-##################################################
-# Override NVIDIA mistaken env var for 11.8 images
-##################################################
-ARG NVIDIA_REQUIRE_CUDA_OVERRIDE
-ENV NVIDIA_REQUIRE_CUDA=${NVIDIA_REQUIRE_CUDA_OVERRIDE:-$NVIDIA_REQUIRE_CUDA}
-
-# Echo the new env var
-RUN echo "NVIDIA_REQUIRE_CUDA_OVERRIDE=${NVIDIA_REQUIRE_CUDA_OVERRIDE}"
-RUN echo "NVIDIA_REQUIRE_CUDA=${NVIDIA_REQUIRE_CUDA}"