From a124eb21686f3dabd85bf53dd6bb9f8e1897f6b3 Mon Sep 17 00:00:00 2001 From: Jason Antic Date: Sun, 2 Jun 2024 15:10:36 -0400 Subject: [PATCH] Fixing mkl version issue causing runtime error when attempting to set gpu (issue #504) --- environment.yml | 1 + 1 file changed, 1 insertion(+) diff --git a/environment.yml b/environment.yml index 56b3625..ef3d378 100644 --- 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