Expand description
Megatron-LM installer — ports scripts/install_megatron.sh.
Constructs correct git clone URL/branch and pip install commands. Megatron depends on PyTorch (ROCm-enabled) and MPI4Py.
§Validation Assertions
- VAL-INSTALL-012: Megatron correct git clone
- VAL-INSTALL-013: Megatron correct pip install post-clone
- VAL-INSTALL-041: Megatron declares dependency on PyTorch and MPI4Py
- VAL-INSTALL-043: DeepSpeed declares dependency on PyTorch
Structs§
- Megatron
Config - Configuration for the Megatron-LM installer.
- Megatron
Installer - The Megatron-LM installer.
- Shell
Command - A constructed shell command.
Enums§
- Install
Method - Installation method.