From 0bea43dba83e11d9d863805f6a702f4709a8c49f Mon Sep 17 00:00:00 2001 From: George Panchuk Date: Wed, 2 Nov 2022 16:13:46 +0400 Subject: [PATCH] fix: fix torch==1.13.0 installation on macosx --- pyproject.toml | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index babbb488..5b4b2bf4 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -14,7 +14,10 @@ keywords = ["framework", "similarity-learning", "metric-learning", "similarity", [tool.poetry.dependencies] python = ">=3.8,<3.11" -torch = ">=1.8.2" +torch = [ + {version = ">=1.8.2", markers = "sys_platform != 'darwin'"}, + {version = ">=1.8.2,<1.13.0", markers = "sys_platform == 'darwin'"} # crashes on torch==1.13.0, tries to install nvidia modules +] pytorch-lightning = "^1.6.4" quaterion-models = "0.1.17" loguru = "^0.5.3"