From 8187f596aaaa303ea0798c6772488d6fcfee1f83 Mon Sep 17 00:00:00 2001 From: bachvudinh Date: Sun, 11 Aug 2024 05:08:58 +0700 Subject: [PATCH] add new base docker images and # Set the MKL_SERVICE_FORCE_INTEL environment variable --- .github/runners/Dockerfile | 5 ++++- lm-evaluation-harness | 2 +- 2 files changed, 5 insertions(+), 2 deletions(-) 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