Skip to content

Commit 2380cab

Browse files
authored
Upgrade torch to 2.12 (Nerogar#1392)
1 parent 43821f8 commit 2380cab

3 files changed

Lines changed: 10 additions & 10 deletions

File tree

requirements-cuda.txt

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,10 @@
11
# pytorch
2-
--extra-index-url https://download.pytorch.org/whl/cu128
3-
torch==2.9.1+cu128
4-
torchvision==0.24.1+cu128
2+
--extra-index-url https://download.pytorch.org/whl/cu130
3+
torch==2.12.0+cu130
4+
torchvision==0.27.0+cu130
55
onnxruntime-gpu==1.23.2
6-
nvidia-nccl-cu12==2.27.5; sys_platform == "linux"
7-
triton-windows==3.5.1.post24; sys_platform == "win32"
6+
nvidia-nccl-cu13==2.29.7; sys_platform == "linux"
7+
triton-windows==3.7.0.post26; sys_platform == "win32"
88

99
# optimizers
1010
bitsandbytes==0.49.1 # bitsandbytes for 8-bit optimizers and weight quantization

requirements-default.txt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
# pytorch
2-
torch==2.9.1
3-
torchvision==0.24.1
2+
torch==2.12.0
3+
torchvision==0.27.0
44
onnxruntime==1.23.2
55

66
# optimizers

requirements-rocm.txt

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2,9 +2,9 @@
22
# please open an issue or pull request on github
33

44
# pytorch
5-
--extra-index-url https://download.pytorch.org/whl/rocm6.3
6-
torch==2.9.1+rocm6.3
7-
torchvision==0.24.1+rocm6.3
5+
--extra-index-url https://download.pytorch.org/whl/rocm7.2
6+
torch==2.12.0+rocm7.2; sys_platform == "linux"
7+
torchvision==0.27.0+rocm7.2; sys_platform == "linux"
88
onnxruntime==1.23.2
99

1010
# optimizers

0 commit comments

Comments
 (0)