Fix CUDA_HOME env var

Kaggle · Dec 17, 2024 · 00192f2 · 00192f2
1 parent 2a33c9f
commit 00192f2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/Dockerfile.tmpl b/Dockerfile.tmpl
@@ -202,7 +202,7 @@ ENV GIT_COMMIT=${GIT_COMMIT} \
 RUN echo "$GIT_COMMIT" > /etc/git_commit && echo "$BUILD_DATE" > /etc/build_date
 
 {{ if eq .Accelerator "gpu" }}
-    # Add the CUDA home.
-    CUDA_HOME=/usr/local/cuda
+# Add the CUDA home.
+ENV CUDA_HOME=/usr/local/cuda
 {{ end }}
 ENTRYPOINT ["/usr/bin/env"]