Skip to main content

Module deepspeed

Module deepspeed 

Source
Expand description

DeepSpeed installer — ports scripts/install_deepspeed.sh.

Constructs correct pip install with ROCm build flags. DeepSpeed depends on PyTorch (validated via preflight check).

§Validation Assertion

  • VAL-INSTALL-006: DeepSpeed installer correct pip command

Structs§

DeepSpeedConfig
Configuration for the DeepSpeed installer.
DeepSpeedInstaller
The DeepSpeed installer.
PipCommand
A constructed pip command.

Enums§

InstallMethod
Installation method.