Skip to content

Commit 8bd3805

Browse files
authored
Fix LD_LIBRARY_PATH for the CPU image. (#1102)
Removing the CUDA stubs from the LD_LIBRARY_PATH should only be done for the GPU build. LD_LIBRARY_PATH_NO_STUBS is not defined for the CPU build causing the LD_LIBRARY_PATH to be set to 0.
1 parent bcc29fb commit 8bd3805

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

Dockerfile.tmpl

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -574,5 +574,7 @@ LABEL kaggle-lang=python
574574
# Correlate current release with the git hash inside the kernel editor by running `!cat /etc/git_commit`.
575575
RUN echo "$GIT_COMMIT" > /etc/git_commit && echo "$BUILD_DATE" > /etc/build_date
576576

577+
{{ if eq .Accelerator "gpu" }}
577578
# Remove the CUDA stubs.
578-
ENV LD_LIBRARY_PATH="$LD_LIBRARY_PATH_NO_STUBS"
579+
ENV LD_LIBRARY_PATH="$LD_LIBRARY_PATH_NO_STUBS"
580+
{{ end }}

0 commit comments

Comments
 (0)