Fixing mkl version issue causing runtime error when attempting to set…

… gpu (issue #504)
jantic · Jun 2, 2024 · a124eb2 · a124eb2
1 parent be725ca
commit a124eb2
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/environment.yml b/environment.yml
@@ -6,6 +6,7 @@ channels:
 dependencies:
 - pip
 - fastai=1.0.60
+- mkl=2024.0
 - python=3.10
 - pytorch::pytorch=1.11.0
 - pytorch::torchvision