separate linux and windows dependencies

mikel-brostrom · May 12, 2024 · fbf11c6 · fbf11c6
1 parent e6d9031
commit fbf11c6
Show file tree

Hide file tree

Showing 2 changed files with 48 additions and 39 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -42,21 +42,13 @@ ftfy = "^6.1.3"
 gitpython = "^3.1.42"
 torch = [
     {version = "^2.2.1", source="pypi", markers = "sys_platform == 'darwin'"},
-    {version = "^2.2.1+cu113", markers = "sys_platform == 'linux'"},
+    {version = "^2.2.1", source="torch_cuda121", markers = "sys_platform == 'linux'"},
+    {version = "^2.2.1", source="torch_cuda121", markers = "sys_platform == 'win32'"},
 ]
 torchvision = [
     {version = "^0.17.1", source="pypi", markers = "sys_platform == 'darwin'"},
-    {version = "^0.17.1+cu113", markers = "sys_platform == 'linux'"},
-]
-
-[tool.poetry.group.cpu.dependencies]
-torch = [
-    {version = "^2.2.1", source="pypi", markers = "sys_platform == 'darwin'"},
-    {version = "^2.2.1", source = "torchcpu", markers = "sys_platform == 'linux'"},
-]
-torchvision = [
-    {version = "^0.17.1", source="pypi", markers = "sys_platform == 'darwin'"},
-    {version = "^0.17.1", source = "torchcpu", markers = "sys_platform == 'linux'"},
+    {version = "^0.17.1", source="torch_cuda121", markers = "sys_platform == 'linux'"},
+    {version = "^0.17.1", source="torch_cuda121", markers = "sys_platform == 'win32'"},
 ]
 
 [tool.poetry.group.test.dependencies]
@@ -86,6 +78,11 @@ name = "torchcpu"
 url = "https://download.pytorch.org/whl/cpu"
 priority = "explicit"
 
+[[tool.poetry.source]]
+name = "torch_cuda121"
+url = "https://download.pytorch.org/whl/cu121"
+priority = "explicit"
+
 
 [build-system]
 requires = ["poetry-core"]