diff --git a/Dockerfile.base b/Dockerfile.base index d1a1eec..dd507f9 100644 --- a/Dockerfile.base +++ b/Dockerfile.base @@ -2,7 +2,16 @@ FROM pytorch/pytorch:2.8.0-cuda12.6-cudnn9-runtime # Install system dependencies -RUN apt update && apt install -y libgl1 && rm -rf /var/lib/apt/lists/* +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 \ + && rm -rf /var/lib/apt/lists/* # Copy and install base requirements (ML dependencies that rarely change) COPY requirements.base.txt .