Skip to content

Commit 2a33c9f

Browse files
authored
Fix LD_LIBRARY_PATH
This was overwriting it to be empty which was breaking GPU function.
1 parent c6e4087 commit 2a33c9f

File tree

1 file changed

+0
-2
lines changed

1 file changed

+0
-2
lines changed

Dockerfile.tmpl

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -202,8 +202,6 @@ ENV GIT_COMMIT=${GIT_COMMIT} \
202202
RUN echo "$GIT_COMMIT" > /etc/git_commit && echo "$BUILD_DATE" > /etc/build_date
203203

204204
{{ if eq .Accelerator "gpu" }}
205-
# Remove the CUDA stubs.
206-
ENV LD_LIBRARY_PATH="$LD_LIBRARY_PATH_NO_STUBS" \
207205
# Add the CUDA home.
208206
CUDA_HOME=/usr/local/cuda
209207
{{ end }}

0 commit comments

Comments
 (0)