cpu specific in poetry toml

openclimatefix · Feb 27, 2024 · 9a31579 · 9a31579
1 parent 9bec505
commit 9a31579
Show file tree

Hide file tree

Showing 2 changed files with 52 additions and 82 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -14,7 +14,7 @@ pvnet = "3.0.5"
 pytz = "^2024.1"
 numpy = "^1.26.4"
 huggingface-hub = "0.20.3"
-torch = "2.2.1"
+torch = {version = "2.1.1", source = "pytorch-cpu" }
 ocf-datapipes = "^3.2.9"
 
 [tool.poetry.group.dev.dependencies]
@@ -36,6 +36,17 @@ name = "pytorch"
 url = "https://download.pytorch.org/whl/cpu"
 priority = "explicit"
 
+[[tool.poetry.source]]
+name = "torch-cpu"
+url = "https://download.pytorch.org/whl/cpu"
+priority = "supplemental"
+
+[[tool.poetry.source]]
+name = "pytorch-cpu"
+url = "https://download.pytorch.org/whl/cpu"
+priority = "explicit"
+
+
 [tool.ruff]
 # Enable pycodestyle (`E`) and Pyflakes (`F`) codes by default.
 select = ["B", "E", "F", "D", "I"]