Skip to main content

Module pytorch

rusty_stack::installers::components

Module pytorch 

Source
Expand description

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.