File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 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
55onnxruntime-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
1010bitsandbytes==0.49.1 # bitsandbytes for 8-bit optimizers and weight quantization
Original file line number Diff line number Diff line change 11# pytorch
2- torch==2.9.1
3- torchvision==0.24.1
2+ torch==2.12.0
3+ torchvision==0.27.0
44onnxruntime==1.23.2
55
66# optimizers
Original file line number Diff line number Diff line change 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"
88onnxruntime==1.23.2
99
1010# optimizers
You can’t perform that action at this time.
0 commit comments