Skip to content

Commit a4cc4b4

Browse files
authored
Add /opt/bin to GPU image PATH env variable (#1089)
NVIDIA binaries from the host are mounted to /opt/bin. This is how `nvidia-smi` is made accessible inside the user session docker container. See attached bugs for additional context. b/202852505
1 parent 0b9588d commit a4cc4b4

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

Diff for: Dockerfile.tmpl

+2
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,8 @@ FROM gcr.io/kaggle-images/python-torch-whl:${GPU_BASE_IMAGE_NAME}-${BASE_IMAGE_T
1414
FROM ${BASE_IMAGE_REPO}/${GPU_BASE_IMAGE_NAME}:${BASE_IMAGE_TAG}
1515
ENV CUDA_MAJOR_VERSION=11
1616
ENV CUDA_MINOR_VERSION=0
17+
# NVIDIA binaries from the host are mounted to /opt/bin.
18+
ENV PATH=/opt/bin:${PATH}
1719
{{ else }}
1820
FROM ${BASE_IMAGE_REPO}/${CPU_BASE_IMAGE_NAME}:${BASE_IMAGE_TAG}
1921
{{ end }}

0 commit comments

Comments
 (0)