diff --git a/.github/runners/Dockerfile b/.github/runners/Dockerfile index 6f5827d..fe31363 100644 --- 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 index bb5da65..fcaa0b1 160000 --- a/lm-evaluation-harness +++ b/lm-evaluation-harness @@ -1 +1 @@ -Subproject commit bb5da65ba11ed26476b5f1d281823d2f643ec69b +Subproject commit fcaa0b1ea72f8ab855eed5ce8bfe07605f110f61