Skip to content

Commit 60735f9

Browse files
committed
Update pytorch to 2.0 and above
1 parent 332898e commit 60735f9

File tree

2 files changed

+95
-123
lines changed

2 files changed

+95
-123
lines changed

poetry.lock

Lines changed: 88 additions & 116 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

pyproject.toml

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -34,8 +34,8 @@ hydra-core = "^1.2.0"
3434

3535
# Optional dependencies
3636
monai = { version = "^1.3.0", extras = ["itk"], optional = true }
37-
torch = { version = "^1.11.0", optional = true }
38-
torchvision = { version = "^0.14.0", optional = true }
37+
torch = { version = "^2.0.0", optional = true }
38+
torchvision = { version = "^0.15.0", optional = true }
3939
torchxrayvision = { version = "^1.2.0", optional = true }
4040
xgboost = { version = "^1.5.2", optional = true }
4141
alibi = { version = "^0.9.4", optional = true, extras = ["shap"] }
@@ -52,26 +52,26 @@ xgboost = "^1.5.2"
5252
optional = true
5353

5454
[tool.poetry.group.torch.dependencies]
55-
torch = "^1.11.0"
55+
torch = "^2.0.0"
5656

5757
[tool.poetry.group.torchvision]
5858
optional = true
5959

6060
[tool.poetry.group.torchvision.dependencies]
61-
torchvision = "^0.14.0"
61+
torchvision = "^0.15.0"
6262

6363
[tool.poetry.group.torchxrayvision]
6464
optional = true
6565

6666
[tool.poetry.group.torchxrayvision.dependencies]
67-
torch = "^1.11.0"
67+
torch = "^2.0.0"
6868
torchxrayvision = "^1.2.0"
6969

7070
[tool.poetry.group.monai]
7171
optional = true
7272

7373
[tool.poetry.group.monai.dependencies]
74-
torch = "^1.11.0"
74+
torch = "^2.0.0"
7575
monai = "^1.1.0"
7676

7777
[tool.poetry.group.alibi]
@@ -86,7 +86,7 @@ optional = true
8686

8787
[tool.poetry.group.alibi-detect.dependencies]
8888
llvmlite = "^0.40.0"
89-
torch = "^1.11.0"
89+
torch = "^2.0.0"
9090
alibi-detect = { version = "^0.12.0", extras = ["torch"] }
9191

9292
[tool.poetry.group.test]

0 commit comments

Comments
 (0)