Struct PyTorchInstaller 

pub struct PyTorchInstaller { /* private fields */ }

The PyTorch installer.

Implementations

impl PyTorchInstaller

pub fn new(config: PyTorchConfig) -> Self

Create a new PyTorch installer with the given config.

pub fn with_defaults() -> Self

Create with default config.

pub fn index_url_for_rocm(&self, rocm_mm: &str) -> String

Get the PyTorch index URL for a given ROCm major.minor version.

Matches the logic in the original script:

• ROCm 7.x: tries Radeon manylinux first, then PyTorch nightly
• ROCm 6.4+: uses PyTorch nightly builds
• ROCm 6.3: uses stable builds
• ROCm 6.0-6.2: uses stable builds for 6.2
• ROCm 5.x: uses stable builds for 5.7
• Fallback: ROCm 6.3 stable

pub fn nightly_index_url(&self, rocm_mm: &str) -> String

Get the PyTorch nightly index URL for a given ROCm version.

pub fn radeon_index_url(&self, rocm_mm: &str) -> String

Get the Radeon manylinux index URL.

pub fn fallback_index_url(&self) -> String

Get the fallback index URL (ROCm 7.0 stable).

pub fn build_install_command( &self, rocm_mm: &str, use_uv: bool, target_version: Option<&str>, ) -> PipCommand

Construct pip install command for PyTorch with ROCm support.

This matches the original script’s pip install logic:

• Uses uv pip install if uv is available
• Uses python3 -m pip install as fallback
• Adds --index-url for the correct ROCm wheel index
• Adds --break-system-packages for global installs
• Installs torch, torchvision, torchaudio

target_version (the planner/manifest target) pins torch to an exact version + --upgrade for update-driven installs, so pip actually advances the installed wheel instead of no-op’ing on an already-satisfied one. A None/empty target (fresh TUI install) leaves torch unpinned → latest. PEP 440: torch==2.13.0 matches the ROCm wheel 2.13.0+rocm7.2, so the pin resolves on the PyTorch ROCm index.

pub fn build_fallback_command(&self, use_uv: bool) -> PipCommand

Construct the fallback command when primary install fails. Tries the ROCm 7.0 stable index.

pub fn build_common_deps_command(&self, use_uv: bool) -> PipCommand

Construct the common ML dependencies install command.

The original script installs torchsde and sentencepiece after PyTorch.