diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index 2fe3827008..8fc06e6058 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -34,6 +34,8 @@ jobs:
       - uses: actions/checkout@v2
         with:
           submodules: 'recursive'
+      - name: Install local Python packages
+        run: pip install .
       - name: Build Software
         working-directory: target/snitch_cluster
         run: |
@@ -66,6 +68,8 @@ jobs:
       - uses: actions/checkout@v2
         with:
           submodules: 'recursive'
+      - name: Install local Python packages
+        run: pip install .
       - name: Build Software
         run: |
           bender vendor init
diff --git a/.github/workflows/lint.yml b/.github/workflows/lint.yml
index 023c20fd80..a2ab4fabce 100644
--- a/.github/workflows/lint.yml
+++ b/.github/workflows/lint.yml
@@ -48,6 +48,8 @@ jobs:
       - uses: actions/checkout@v4
         with:
           submodules: true
+      - name: Install local Python packages
+        run: pip install .
       - name: Generate opcodes
         run: |
           ./util/generate-opcodes.sh
diff --git a/util/container/Dockerfile b/util/container/Dockerfile
index 0bbd03730e..f6e98f7135 100644
--- a/util/container/Dockerfile
+++ b/util/container/Dockerfile
@@ -155,10 +155,10 @@ ENV VIRTUAL_ENV "/root/.venvs/snitch_cluster"
 RUN /opt/python/bin/python3 -m venv ${VIRTUAL_ENV}
 ENV PATH="${VIRTUAL_ENV}/bin:${PATH}"
 # Install Python requirements
-COPY . /tmp/snitch_cluster
-RUN pip install -r /tmp/snitch_cluster/python-requirements.txt
-RUN pip install /tmp/snitch_cluster/
-RUN rm -rf /tmp/snitch_cluster
+COPY python-requirements.txt /tmp/python-requirements.txt
+COPY docs/requirements.txt /tmp/docs/requirements.txt
+COPY sw/dnn/requirements.txt /tmp/sw/dnn/requirements.txt
+RUN pip install -r /tmp/python-requirements.txt
 
 # Add Verilator to PATH
 ENV PATH "/tools/verilator/bin:${PATH}"