diff --git a/runner/docker/Dockerfile.live-base-comfyui b/runner/docker/Dockerfile.live-base-comfyui
index 66ede573..04d24ad2 100644
--- a/runner/docker/Dockerfile.live-base-comfyui
+++ b/runner/docker/Dockerfile.live-base-comfyui
@@ -35,14 +35,14 @@ RUN cd /comfyui/custom_nodes && \
     cd ComfyUI-Misc-Effects && \
     git checkout c6b360c78611134c3723388170475eb4898ff6b7
 
-RUN pip install torch==2.5.1 torchvision torchaudio tqdm nvidia-ml-py==12.560.30
+RUN pip install torch==2.5.1 torchvision torchaudio tqdm
 
 # Install comfystream (which includes ComfyUI)
 RUN pip install git+https://github.com/yondonfu/comfystream.git
 RUN git clone https://github.com/yondonfu/comfystream.git && \
     cd comfystream && \
     pip install -r requirements.txt && \
-    cp -r nodes/tensor_utils /comfyui/custom_nodes/ 
+    cp -r nodes/tensor_utils /comfyui/custom_nodes/
 
 # Install ComfyUI-SAM2-Realtime (https://github.com/pschroedl/ComfyUI-SAM2-Realtime.git)
 RUN cd /comfyui/custom_nodes && \
diff --git a/runner/docker/Dockerfile.live-base-streamdiffusion b/runner/docker/Dockerfile.live-base-streamdiffusion
index de4cf897..621a80c2 100644
--- a/runner/docker/Dockerfile.live-base-streamdiffusion
+++ b/runner/docker/Dockerfile.live-base-streamdiffusion
@@ -21,8 +21,7 @@ RUN pip install --no-cache-dir \
 RUN pip install --no-cache-dir \
     huggingface-hub==0.23.2 \
     diffusers==0.30.0 \
-    protobuf==5.27.2 \
-    nvidia-ml-py==12.560.30 
+    protobuf==5.27.2
 
 # Install StreamDiffusion @ 765d71029b1404b94aee2865178d71c257c20318 (latest at time of writing)
 RUN pip install git+https://github.com/cumulo-autumn/StreamDiffusion.git@765d710#egg=streamdiffusion[tensorrt]
diff --git a/runner/requirements-liveportrait.txt b/runner/requirements-liveportrait.txt
index c1150ee2..afac691a 100644
--- a/runner/requirements-liveportrait.txt
+++ b/runner/requirements-liveportrait.txt
@@ -10,4 +10,4 @@ pycuda==2024.1.2
 scikit-image==0.24.0
 torchgeometry==0.1.2
 torchvision==0.19.1
-nvidia-ml-py==12.560.30
+nvidia-ml-py==12.560.30  # Includes pynvml.
diff --git a/runner/requirements.live-ai.txt b/runner/requirements.live-ai.txt
index a851a269..75a83c46 100644
--- a/runner/requirements.live-ai.txt
+++ b/runner/requirements.live-ai.txt
@@ -22,4 +22,4 @@ PyGObject==3.50.0
 onnxruntime==1.19.2
 watchdog==5.0.2
 aiohttp==3.10.9
-nvidia-ml-py==12.560.30
\ No newline at end of file
+nvidia-ml-py==12.560.30  # Includes pynvml.
diff --git a/runner/requirements.txt b/runner/requirements.txt
index 17237444..9998941a 100644
--- a/runner/requirements.txt
+++ b/runner/requirements.txt
@@ -20,5 +20,4 @@ protobuf==5.27.2
 bitsandbytes==0.43.3
 psutil==6.0.0
 PyYAML==6.0.2
-nvidia-ml-py==12.560.30
-pynvml==12.0.0
+nvidia-ml-py==12.560.30  # Includes pynvml.