Add unofficial TE wheel built from vuiseng9's fork: branch v2.5+nvfp4 (010e0427)
Browse files
readme.md
CHANGED
|
@@ -3,6 +3,7 @@ Transformer Engine prebuilt Wheel
|
|
| 3 |
|
| 4 |
Targeting >= Hopper
|
| 5 |
|
|
|
|
| 6 |
```
|
| 7 |
250823_a/transformer_engine-2.5.0+f05f12c9-cp312-cp312-linux_x86_64.whl
|
| 8 |
* build environment: nvcr.io/nvidia/pytorch:25.06-py3
|
|
@@ -10,4 +11,15 @@ Targeting >= Hopper
|
|
| 10 |
* cuda : 12.9
|
| 11 |
* cudnn: 91002 (9.10.2)
|
| 12 |
* TE stable v2.5 (CUDA_ARCHS="90;100;120")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 13 |
```
|
|
|
|
| 3 |
|
| 4 |
Targeting >= Hopper
|
| 5 |
|
| 6 |
+
## Official Release, prebuilt wheel
|
| 7 |
```
|
| 8 |
250823_a/transformer_engine-2.5.0+f05f12c9-cp312-cp312-linux_x86_64.whl
|
| 9 |
* build environment: nvcr.io/nvidia/pytorch:25.06-py3
|
|
|
|
| 11 |
* cuda : 12.9
|
| 12 |
* cudnn: 91002 (9.10.2)
|
| 13 |
* TE stable v2.5 (CUDA_ARCHS="90;100;120")
|
| 14 |
+
```
|
| 15 |
+
|
| 16 |
+
## Unofficial, prebuilt wheel for local experimentation
|
| 17 |
+
```
|
| 18 |
+
unofficial/251015_a/transformer_engine-2.5.0+010e0427-cp312-cp312-linux_x86_64.whl
|
| 19 |
+
* build environment: nvcr.io/nvidia/pytorch:25.06-py3
|
| 20 |
+
* torch: 2.8.0a0+5228986c39.nv25.06
|
| 21 |
+
* cuda : 12.9
|
| 22 |
+
* cudnn: 91002 (9.10.2)
|
| 23 |
+
* TE: vuiseng9's fork: branch v2.5+nvfp4 (010e0427)
|
| 24 |
+
# CUDA_ARCHS="100"
|
| 25 |
```
|
unofficial/251015_a/transformer_engine-2.5.0+010e0427-cp312-cp312-linux_x86_64.whl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1aede90d2a7376ecfb0352b43e4f109f31d67aedcb3ac62262a010dbff3d77a5
|
| 3 |
+
size 150541317
|