How to install transformer-engine?Which version of transformer-engine-torch?

[KivenJoo]

When installing transformer_engine, the version is always wrong. Does the version of transformer_engine_torch conflict with the version of pytorch?

[KivenJoo]

When installing transformer_engine, the version is always wrong. Does the version of transformer_engine_torch conflict with the version of pytorch?

And is this version also related to the cuda version? What version of cuda should I use to match this version?

[orchidmalevolence]

We test Ming-Flash-Omni on cuda12.4 and transformer-engine==1.11.0

[sbhavani]

@JokerColyn please see https://github.com/NVIDIA/TransformerEngine?tab=readme-ov-file#installation