From e5578555582e3cc44d871ca631d1b9e8e7d7eeb2 Mon Sep 17 00:00:00 2001 From: Mohit Sharma Date: Tue, 6 Aug 2024 12:46:35 +0000 Subject: [PATCH] revert mkl --- Dockerfile_amd | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Dockerfile_amd b/Dockerfile_amd index 129a58fc..d8bb8c47 100644 --- a/Dockerfile_amd +++ b/Dockerfile_amd @@ -92,7 +92,7 @@ RUN chmod +x ~/mambaforge.sh && \ # Install flash-attention, torch dependencies RUN pip install numpy einops ninja --no-cache-dir -RUN conda install intel::mkl-static intel::mkl-include +RUN conda install mkl-static mkl-include RUN pip uninstall -y triton && \ git clone --depth 1 --single-branch https://github.com/ROCm/triton.git && \ cd triton/python && \