avoid mkl-pytorch bug

bioimage-io · May 29, 2024 · 1f36013 · 1f36013
1 parent 7e708c7
commit 1f36013
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/bioimageio_collection_backoffice/validate_format.py b/bioimageio_collection_backoffice/validate_format.py
@@ -146,7 +146,11 @@ def get_pytorch_env(
     conda_env = get_base_env()
     conda_env["channels"].insert(0, "pytorch")
     conda_env["dependencies"].extend(
-        [f"pytorch {get_version_range(pytorch_version)}", "cpuonly"]
+        [
+            f"pytorch {get_version_range(pytorch_version)}",
+            "cpuonly",
+            "mkl !=2024.1.0",  #  avoid https://github.com/pytorch/pytorch/issues/123097
+        ]
     )
     return conda_env