Remove CUDA stubs at the end of the GPU build · JasonYangCode/docker-python@01dcc9d · GitHub

JasonYangCode / docker-python Public

forked from Kaggle/docker-python

• Notifications You must be signed in to change notification settings
• Fork 0
• Star 1

• Code
• Pull requests
• Actions
• Projects
• Security and quality
• Insights

Additional navigation options

Commit 01dcc9d

Browse filesBrowse the repository at this point in the historyBrowse files

rosbo

committed

Remove CUDA stubs at the end of the GPU build

1 parent 4126b95 commit 01dcc9d
Copy full SHA for 01dcc9d

1 file changed

+4Lines changed: 4 additions & 0 deletions

File tree

Expand file treeCollapse file tree

Open diff view settings

Filter options

• gpu.Dockerfile

Expand file treeCollapse file tree

Open diff view settings

Collapse file

‎gpu.Dockerfile‎

Copy file name to clipboardExpand all lines: gpu.Dockerfile

+4Lines changed: 4 additions & 0 deletions

Original file line number	Diff line number	Diff line change
@@ -28,6 +28,7 @@ ENV PATH=/usr/local/nvidia/bin:/usr/local/cuda/bin:${PATH}
28	28	# CUDA user libraries, either manually or through the use of nvidia-docker) exclude them. One
29	29	# convenient way to do so is to obscure its contents by a bind mount:
30	30	# docker run .... -v /non-existing-directory:/usr/local/cuda/lib64/stubs:ro ...
	31	+ ENV LD_LIBRARY_PATH_NO_STUBS="/usr/local/nvidia/lib64:/usr/local/cuda/lib64:$LD_LIBRARY_PATH"
31	32	ENV LD_LIBRARY_PATH="/usr/local/nvidia/lib64:/usr/local/cuda/lib64:/usr/local/cuda/lib64/stubs:$LD_LIBRARY_PATH"
32	33	ENV NVIDIA_VISIBLE_DEVICES=all
33	34	ENV NVIDIA_DRIVER_CAPABILITIES=compute,utility
@@ -101,3 +102,6 @@ RUN pip install pycuda && \
101	102	# Re-add TensorBoard Jupyter extension patch
102	103	# b/139212522 re-enable TensorBoard once solution for slowdown is implemented.
103	104	# ADD patches/tensorboard/notebook.py /opt/conda/lib/python3.7/site-packages/tensorboard/notebook.py
	105	+
	106	+ # Remove the CUDA stubs.
	107	+ ENV LD_LIBRARY_PATH="$LD_LIBRARY_PATH_NO_STUBS"

0 commit comments

Comments

0 (0)