File tree Expand file tree Collapse file tree 1 file changed +9
-2
lines changed Expand file tree Collapse file tree 1 file changed +9
-2
lines changed Original file line number Diff line number Diff line change @@ -22,8 +22,15 @@ onnxruntime = "^1.13.1"
22
22
onnxruntime-directml = " ^1.15.0"
23
23
colorama = " ^0.4.6"
24
24
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
+
27
34
[tool .poetry .group .dev .dependencies ]
28
35
black = " ^22.10.0"
29
36
pyinstaller = " ^5.6.2"
You can’t perform that action at this time.
0 commit comments