OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

Dockerfile; Add cuda bin to PATH (#117105)

Browse Source 
We need this to execute `nvidia-smi` in the officially released containers. We have already it in the Docker CI

See
94db6578cc/.ci/docker/linter-cuda/Dockerfile (L35)

Pull Request resolved: https://github.com/pytorch/pytorch/pull/117105
Approved by: https://github.com/atalman
This commit is contained in:
bhack 2024-01-11 18:10:14 +00:00 committed by PyTorch MergeBot
parent 9e3580f793
commit 5bc896e5dc
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
Dockerfile
View File

@ -99,6 +99,7 @@ ENV PATH /opt/conda/bin:$PATH
ENV NVIDIA_VISIBLE_DEVICES all ENV NVIDIA_VISIBLE_DEVICES all
ENV NVIDIA_DRIVER_CAPABILITIES compute,utility ENV NVIDIA_DRIVER_CAPABILITIES compute,utility
ENV LD_LIBRARY_PATH /usr/local/nvidia/lib:/usr/local/nvidia/lib64 ENV LD_LIBRARY_PATH /usr/local/nvidia/lib:/usr/local/nvidia/lib64
ENV PATH /usr/local/nvidia/bin:/usr/local/cuda/bin:$PATH
ENV PYTORCH_VERSION ${PYTORCH_VERSION} ENV PYTORCH_VERSION ${PYTORCH_VERSION}
WORKDIR /workspace WORKDIR /workspace