add new base docker images and # Set the MKL_SERVICE_FORCE_INTEL envi…

…ronment variable
janhq · Aug 10, 2024 · 8187f59 · 8187f59
1 parent 3bd0d3f
commit 8187f59
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/.github/runners/Dockerfile b/.github/runners/Dockerfile
@@ -1,4 +1,7 @@
-FROM docker.io/pytorch/pytorch:2.3.0-cuda12.1-cudnn8-runtime
+FROM docker.io/pytorch/pytorch:2.4.0-cuda12.1-cudnn9-runtime
+
+# Set the MKL_SERVICE_FORCE_INTEL environment variable
+ENV MKL_SERVICE_FORCE_INTEL=1
 
 RUN apt-get update \
   && apt-get install -y --no-install-recommends \

diff --git a/lm-evaluation-harness b/lm-evaluation-harness