diff --git a/devops/develop/Dockerfile b/devops/develop/Dockerfile
index 73f8e0f22..1d4c36ecd 100644
--- a/devops/develop/Dockerfile
+++ b/devops/develop/Dockerfile
@@ -11,11 +11,13 @@ RUN /bin/bash -xc '. /virtualenv/env3/bin/activate \
  && pip install --upgrade pip setuptools-scm \
  && pip install --no-cache-dir -r /wbia/wildbook-ia/requirements/tests.txt \
  && pip install --no-cache-dir -r /wbia/wildbook-ia/requirements/runtime.txt \
+ && pip install --no-cache-dir -r /wbia/wildbook-ia/requirements/torch.txt \
+ && pip install --no-cache-dir -r /wbia/wildbook-ia/requirements/runtime-post.txt \
  && pip install --no-cache-dir -r /wbia/wildbook-ia/requirements/postgres.txt \
  && pip install --no-cache-dir -e .'
 
 # Temporary fix for ARM64
-ENV LD_PRELOAD "/virtualenv/env3/lib/python3.7/site-packages/torch/lib/libgomp-d22c30c5.so.1"
+ENV LD_PRELOAD "/virtualenv/env3/lib/python3.8/site-packages/torch/lib/libgomp-d22c30c5.so.1"
 
 RUN set -ex \