From 5e696956d7a62216e36a18793aca2267ea53f3ec Mon Sep 17 00:00:00 2001 From: carlosuc3m <100329787@alumnos.uc3m.es> Date: Tue, 16 Apr 2024 12:05:59 +0200 Subject: [PATCH] solve known mkl pytorch error https://github.com/pytorch/pytorch/issues/123097 --- src/main/java/ai/nets/samj/SamEnvManager.java | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/main/java/ai/nets/samj/SamEnvManager.java b/src/main/java/ai/nets/samj/SamEnvManager.java index a7406c1..1729489 100644 --- a/src/main/java/ai/nets/samj/SamEnvManager.java +++ b/src/main/java/ai/nets/samj/SamEnvManager.java @@ -84,7 +84,8 @@ public class SamEnvManager { /** * Dependencies that have to be installed in any SAMJ created environment using Mamba or Conda */ - final public static List INSTALL_CONDA_DEPS = Arrays.asList(new String[] {"libpng", "libjpeg-turbo", "scikit-image", "pytorch=2.0.1", "torchvision=0.15.2", "cpuonly"}); + final public static List INSTALL_CONDA_DEPS = Arrays.asList(new String[] {"libpng", "libjpeg-turbo", + "scikit-image", "pytorch=2.0.1", "torchvision=0.15.2", "cpuonly", "mkl=2024.0.0"}); /** * Dependencies that have to be installed using Mamba or Conda in environments that are going * to be used to run EfficientViTSAM