Adding Cuda home environment values helps other libraries avoid unwan…

…ted errors, for example DeepSpeed (#1360)
Kaggle · Feb 1, 2024 · 71487b6 · 71487b6
1 parent 65e4cd1
commit 71487b6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/Dockerfile.tmpl b/Dockerfile.tmpl
@@ -708,4 +708,6 @@ RUN echo "$GIT_COMMIT" > /etc/git_commit && echo "$BUILD_DATE" > /etc/build_date
 {{ if eq .Accelerator "gpu" }}
 # Remove the CUDA stubs.
 ENV LD_LIBRARY_PATH="$LD_LIBRARY_PATH_NO_STUBS"
+# Add the CUDA home.
+ENV CUDA_HOME=/usr/local/cuda
 {{ end }}