Transformer Engine prebuilt Wheel
Official Release, prebuilt wheel
260121_a/transformer_engine-2.5.0+f05f12c9-cp312-cp312-linux_x86_64.whl
* build environment: nvcr.io/nvidia/pytorch:25.06-py3
* torch: 2.8.0a0+5228986c39.nv25.06
* cuda : 12.9
* cudnn: 91002 (9.10.2)
* TE branch release_v2.11 c188b5 dec 16 (CUDA_ARCHS="90")
250823_a/transformer_engine-2.5.0+f05f12c9-cp312-cp312-linux_x86_64.whl
* build environment: nvcr.io/nvidia/pytorch:25.06-py3
* torch: 2.8.0a0+5228986c39.nv25.06
* cuda : 12.9
* cudnn: 91002 (9.10.2)
* TE stable v2.5 (CUDA_ARCHS="90;100;120")
Unofficial, prebuilt wheel for local experimentation
unofficial/251115_a/transformer_engine-2.5.0+c2056e01-cp312-cp312-linux_x86_64.whl
* build environment: nvcr.io/nvidia/nemo:25.07
torch: 2.8.0a0+5228986c39.nv25.05
cuda : 12.9
cudnn: 91100
* TE vuiseng9's fork:v2.5+nv4f_mx8b (CUDA_ARCHS=100)
unofficial/251025_a/transformer_engine-2.8.0+7cde9a33-cp312-cp312-linux_x86_64.whl
* build environment: nvcr.io/nvidia/nemo:25.09
torch: 2.8.0a0+5228986c39.nv25.06
cuda : 12.9
cudnn: 91100 (9.11.0)
* TE vuiseng9's fork:v2.8 ++ (CUDA_ARCHS=100a)
unofficial/251020_a/transformer_engine-2.5.0+c2056e01-cp312-cp312-linux_x86_64.whl
* build environment: nvcr.io/nvidia/nemo:25.07
* torch: 2.8.0a0+5228986c39.nv25.05
* cuda : 12.9
* cudnn: 91100 (9.11.0)
* TE: vuiseng9's fork: branch v2.5+nvfp4 (c2056e01)
# CUDA_ARCHS="100"
unofficial/251015_a/transformer_engine-2.5.0+010e0427-cp312-cp312-linux_x86_64.whl
* build environment: nvcr.io/nvidia/pytorch:25.06-py3
* torch: 2.8.0a0+5228986c39.nv25.06
* cuda : 12.9
* cudnn: 91002 (9.10.2)
* TE: vuiseng9's fork: branch v2.5+nvfp4 (010e0427)
# CUDA_ARCHS="100"