Update Dockerfile.rocm

Alexei-V-Ivanov-AMD · May 17, 2024 · 2e32cf7 · 2e32cf7
1 parent 48a37af
commit 2e32cf7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)