Skip to content

Commit

Permalink
add sources
Browse files Browse the repository at this point in the history
  • Loading branch information
@narumiruna
narumiruna committed Jun 24, 2024
1 parent a0b7452 commit 92c4f2b
Show file tree
Hide file tree
Showing 2 changed files with 11 additions and 12 deletions.
6 changes: 3 additions & 3 deletions poetry.lock
View file

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

17 changes: 8 additions & 9 deletions pyproject.toml
View file
Original file line number Diff line number Diff line change
Expand Up @@ -18,8 +18,14 @@ priority = "explicit"
python = "^3.11"
# torch = "^2.3.1"
# torchvision = "^0.18.1"
torch = { version = "^2.3.1+cu121", source = "pytorch-cu121", markers = "extra=='cuda' and extra!='cpu'" }
torchvision = { version = "^0.18.1+cu121", source = "pytorch-cu121", markers = "extra=='cuda' and extra!='cpu'" }
torch = [
{ version = "^2.3.1+cu121", source = "pytorch-cu121", markers = "extra=='cuda' and extra!='cpu'" },
{ version = "^2.3.1+cpu", source = "pytorch-cpu", markers = "extra=='cpu' and extra!='cuda'" },
]
torchvision = [
{ version = "^0.18.1+cu121", source = "pytorch-cu121", markers = "extra=='cuda' and extra!='cpu'" },
{ version = "^0.18.1+cpu", source = "pytorch-cpu", markers = "extra=='cpu' and extra!='cuda'" },
]
torchmetrics = "^1.4.0"
numpy = "^1.26.3"
tqdm = "^4.66.4"
Expand All @@ -34,13 +40,6 @@ ruff = "^0.4.10"
toml = "^0.10.2"
mypy = "^1.10.0"

[tool.poetry.group.cpu]
optional = true

[tool.poetry.group.cpu.dependencies]
torch = { version = "^2.3.1+cpu", source = "pytorch-cpu", markers = "extra=='cpu' and extra!='cuda'" }
torchvision = { version = "^0.18.1+cpu", source = "pytorch-cpu", markers = "extra=='cpu' and extra!='cuda'" }

[tool.poetry.extras]
cuda = ["torch", "torchvision"]
cpu = ["torch", "torchvision"]
Expand Down

0 comments on commit 92c4f2b

Please sign in to comment.