Module pytorch 

PyTorch installer — ports scripts/install_pytorch_rocm.sh.

Constructs correct pip install commands with the right index URL for ROCm version + Python version wheel selection.

Validation Assertion

• VAL-INSTALL-003: PyTorch installer correct wheel selection

Structs

PipCommand

A constructed pip install command.

PyTorchConfig

Configuration for the PyTorch installer.

PyTorchInstaller

The PyTorch installer.

Enums

InstallMethod

Installation method.

TorchChannel

PyTorch release channel.

Constants

AMD_WHL_NEXT_INDEX

AMD’s PyTorch wheel index for ROCm 10.x (TheRock) releases.