diff --git a/Dockerfile.base b/Dockerfile.base
index dd507f9..60999b1 100644
--- a/Dockerfile.base
+++ b/Dockerfile.base
@@ -5,12 +5,12 @@ FROM pytorch/pytorch:2.8.0-cuda12.6-cudnn9-runtime
 RUN apt update && apt install -y \
     libgl1 \
     libglib2.0-0 \
-    libgthread-2.0-0 \
     libgstreamer1.0-0 \
-    libgtk2.0-0 \
-    libavcodec-dev \
-    libavformat-dev \
-    libswscale-dev \
+    libgtk-3-0 \
+    libavcodec58 \
+    libavformat58 \
+    libswscale5 \
+    libgomp1 \
     && rm -rf /var/lib/apt/lists/*
 
 # Copy and install base requirements (ML dependencies that rarely change)