diff --git a/environment.yaml b/environment.yaml
index 638b5a2..4382519 100644
--- a/environment.yaml
+++ b/environment.yaml
@@ -9,6 +9,7 @@ dependencies:
   - cudatoolkit=11.8
   - pytorch=2.0.1 
   - torchvision=0.15.2
+  - mkl=2024.0
   - pip
   - pip:
     - absl-py==2.1.0