Skip to content

Commit 04f033b

Browse files
authored
feat: bump torch, torchvision, triton to 2.10.0, 0.25.0, 3.6.0 (#245)
1 parent 03ed06d commit 04f033b

4 files changed

Lines changed: 16 additions & 10 deletions

File tree

.metwork-framework/components.md

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,8 @@
22
| --- | --- | --- |
33
| [absl-py](https://github.com/abseil/abseil-py) | 2.2.2 | python3_ia |
44
| [astunparse](https://github.com/simonpercivall/astunparse) | 1.6.3 | python3_ia |
5+
| [cuda-bindings](https://pypi.org/project/cuda-bindings) | 12.9.4 | python3_ia |
6+
| [cuda-pathfinder](https://pypi.org/project/cuda-pathfinder) | 1.1.0 | python3_ia |
57
| [diffusers](https://github.com/huggingface/diffusers) | 0.33.1 | python3_ia |
68
| [dm-tree](https://github.com/deepmind/tree) | 0.1.9 | python3_ia |
79
| [einops](https://github.com/arogozhnikov/einops) | 0.8.1 | python3_ia |
@@ -32,7 +34,7 @@
3234
| [nvidia-cusparselt-cu12](https://developer.nvidia.com/cusparselt) | 0.7.1 | python3_ia |
3335
| [nvidia-nccl-cu12](https://developer.nvidia.com/cuda-zone) | 2.27.5 | python3_ia |
3436
| [nvidia-nvjitlink-cu12](https://developer.nvidia.com/cuda-zone) | 12.8.93 | python3_ia |
35-
| [nvidia-nvshmem-cu12](https://developer.nvidia.com/cuda-zone) | 3.3.20 | python3_ia |
37+
| [nvidia-nvshmem-cu12](https://developer.nvidia.com/cuda-zone) | 3.4.5 | python3_ia |
3638
| [nvidia-nvtx-cu12](https://developer.nvidia.com/cuda-zone) | 12.8.90 | python3_ia |
3739
| [oauthlib](https://github.com/oauthlib/oauthlib) | 3.2.2 | python3_ia |
3840
| [onnx](https://onnx.ai/) | 1.20.0 | python3_ia |
@@ -53,10 +55,10 @@
5355
| [tf_keras](https://keras.io/) | 2.20.1 | python3_ia |
5456
| [timm](https://github.com/huggingface/pytorch-image-models) | 1.0.22 | python3_ia |
5557
| [tokenizers](https://github.com/huggingface/tokenizers) | 0.21.1 | python3_ia |
56-
| [torch](https://pytorch.org) | 2.9.1 | python3_ia |
58+
| [torch](https://pytorch.org) | 2.10.0 | python3_ia |
5759
| [torchmetrics](https://github.com/Lightning-AI/torchmetrics) | 1.7.1 | python3_ia |
58-
| [torchvision](https://github.com/pytorch/vision) | 0.24.1 | python3_ia |
60+
| [torchvision](https://github.com/pytorch/vision) | 0.25.0 | python3_ia |
5961
| [transformers](https://github.com/huggingface/transformers) | 4.53.3 | python3_ia |
60-
| [triton](https://github.com/triton-lang/triton/) | 3.5.1 | python3_ia |
62+
| [triton](https://github.com/triton-lang/triton/) | 3.6.0 | python3_ia |
6163

62-
*(58 components)*
64+
*(60 components)*
Binary file not shown.

layers/layer5_python3_ia/0500_extra_packages/allow_binary_packages

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,8 @@ absl-py
22
aiohappyeyeballs
33
astunparse
44
cmake
5+
cuda-bindings
6+
cuda-pathfinder
57
flatbuffers
68
google-auth
79
google-auth-oauthlib

layers/layer5_python3_ia/0500_extra_packages/requirements3.txt

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,7 @@
1-
astunparse==1.6.3
21
absl-py==2.2.2
2+
astunparse==1.6.3
3+
cuda-bindings==12.9.4
4+
cuda-pathfinder==1.1.0
35
diffusers==0.33.1
46
dm-tree==0.1.9
57
einops==0.8.1
@@ -30,7 +32,7 @@ nvidia-cusparse-cu12==12.5.8.93
3032
nvidia-cusparselt-cu12==0.7.1
3133
nvidia-nccl-cu12==2.27.5
3234
nvidia-nvjitlink-cu12==12.8.93
33-
nvidia-nvshmem-cu12==3.3.20
35+
nvidia-nvshmem-cu12==3.4.5
3436
nvidia-nvtx-cu12==12.8.90
3537
oauthlib==3.2.2
3638
onnx==1.20.0
@@ -51,8 +53,8 @@ tensorflow-hub==0.16.1
5153
tf-keras==2.20.1
5254
timm==1.0.22
5355
tokenizers==0.21.1
54-
torch==2.9.1
56+
torch==2.10.0
5557
torchmetrics==1.7.1
56-
torchvision==0.24.1
58+
torchvision==0.25.0
5759
transformers==4.53.3
58-
triton==3.5.1
60+
triton==3.6.0

0 commit comments

Comments
 (0)