From 5bc896e5dc856ca831a7f78fdda3b95b1cb8c631 Mon Sep 17 00:00:00 2001
From: bhack <bhack@users.noreply.github.com>
Date: Thu, 11 Jan 2024 18:10:14 +0000
Subject: [PATCH] Dockerfile; Add cuda bin to PATH (#117105)

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

See
https://github.com/pytorch/pytorch/blob/94db6578ccee2551c986d92c245e0a0729b99449/.ci/docker/linter-cuda/Dockerfile#L35

Pull Request resolved: https://github.com/pytorch/pytorch/pull/117105
Approved by: https://github.com/atalman
---
 Dockerfile | 1 +
 1 file changed, 1 insertion(+)

diff --git a/Dockerfile b/Dockerfile
index a2c4bef96598..ff8aa3e5a9c4 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -99,6 +99,7 @@ ENV PATH /opt/conda/bin:$PATH
 ENV NVIDIA_VISIBLE_DEVICES all
 ENV NVIDIA_DRIVER_CAPABILITIES compute,utility
 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}
 WORKDIR /workspace