Struct DeepSpeedInstaller 

pub struct DeepSpeedInstaller { /* private fields */ }

The DeepSpeed installer.

Implementations

impl DeepSpeedInstaller

pub fn new(config: DeepSpeedConfig) -> Self

Create a new DeepSpeed installer with the given config.

pub fn with_defaults() -> Self

Create with default config.

pub fn build_deps_install_command(&self) -> PipCommand

Construct the pip install command for DeepSpeed dependencies.

Installs packaging, ninja, pydantic, jsonschema, einops.

pub fn build_install_command(&self) -> PipCommand

Construct the pip install command for DeepSpeed.

The original script uses pip install deepspeed einops with ROCM_HOME and HIP_PATH set. It does NOT set DS_ACCELERATOR=rocm as that caused ValueError in newer versions.

pub fn build_no_deps_command(&self) -> PipCommand

Construct the retry install command with –no-deps.

Used when the primary install fails.

pub fn build_force_reinstall_command(&self) -> PipCommand

Construct the force-reinstall retry command.

Uses --no-deps so a force-reinstall never pulls/overrides the ROCm torch (Stage 3: never override a core component).

pub fn required_dependencies(&self) -> &[&str]

Get the list of required dependencies to verify after install.