Skip to main content

Module pytorch_profiler

rusty_stack::installers::components

Module pytorch_profiler 

Source
Expand description

PyTorch Profiler installer — ports scripts/install_pytorch_profiler.sh.

Constructs correct pip install command for PyTorch Profiler (torch-tb-profiler) and TensorBoard. Depends on PyTorch being installed first.

§Validation Assertion

  • VAL-INSTALL-020: PyTorch Profiler correct pip command

Structs§

PytorchProfilerConfig
Configuration for the PyTorch Profiler installer.
PytorchProfilerInstaller
The PyTorch Profiler installer.
ShellCommand
A constructed shell command.

Enums§

InstallMethod
Installation method.