Skip to content

Commit 0dca722

Browse files
authored
Merge pull request #1 from Meister1593/main
Changed to cpu torch by default, updated lock
2 parents 931aa19 + e6b1665 commit 0dca722

File tree

1 file changed

+9
-2
lines changed

1 file changed

+9
-2
lines changed

pyproject.toml

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,8 +22,15 @@ onnxruntime = "^1.13.1"
2222
onnxruntime-directml = "^1.15.0"
2323
colorama = "^0.4.6"
2424
taskipy = "^1.10.4"
25-
torch = "~2.0.0"
26-
torchvision = "~0.15.1"
25+
torch = [
26+
{url = "https://download.pytorch.org/whl/cpu/torch-2.0.0%2Bcpu-cp310-cp310-linux_x86_64.whl", markers = "sys_platform == 'linux'"},
27+
{url = "https://download.pytorch.org/whl/cpu/torch-2.0.0%2Bcpu-cp310-cp310-win_amd64.whl", markers = "sys_platform == 'win32'" }
28+
]
29+
torchvision = [
30+
{url = "https://download.pytorch.org/whl/cpu/torchvision-0.15.1%2Bcpu-cp310-cp310-linux_x86_64.whl", markers = "sys_platform == 'linux'"},
31+
{url = "https://download.pytorch.org/whl/cpu/torchvision-0.15.1%2Bcpu-cp310-cp310-win_amd64.whl", markers = "sys_platform == 'win32'" }
32+
]
33+
2734
[tool.poetry.group.dev.dependencies]
2835
black = "^22.10.0"
2936
pyinstaller = "^5.6.2"

0 commit comments

Comments
 (0)