Update Dockerfile

WildMeOrg · Jun 15, 2024 · 428fefd · 428fefd
1 parent 694feb5
commit 428fefd
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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 \