From 2e32cf7627125097bfe9cdebfb1e52ecb3d09c2e Mon Sep 17 00:00:00 2001
From: Alexei-V-Ivanov-AMD
 <156011006+Alexei-V-Ivanov-AMD@users.noreply.github.com>
Date: Fri, 17 May 2024 18:17:54 -0500
Subject: [PATCH] Update Dockerfile.rocm

---
 Dockerfile.rocm | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/Dockerfile.rocm b/Dockerfile.rocm
index a3d9ef20dce..9bfe8446a51 100644
--- a/Dockerfile.rocm
+++ b/Dockerfile.rocm
@@ -92,7 +92,7 @@ RUN if [ "$BUILD_TRITON" = "1" ]; then \
 WORKDIR /vllm-workspace
 COPY . .
 
-RUN python3 -m pip install pynvml # to be removed eventually
+#RUN python3 -m pip install pynvml # to be removed eventually
 RUN python3 -m pip install --upgrade pip numba
 
 # make sure punica kernels are built (for LoRA)