Skip to main content

Module detect

Module detect 

Source
Expand description

libtorch installation detection and .arch metadata parsing.

Structs§

LibtorchInfo
Metadata about an installed libtorch variant (from .arch file).

Functions§

arch_compatible
Check whether a GPU’s compute capability is covered by the libtorch variant’s compiled architectures (from the .arch file).
is_valid_variant
Check whether a libtorch variant directory looks valid (has lib/).
list_variants
List all installed libtorch variants under <root>/libtorch/.
read_active
Read the active libtorch variant from <root>/libtorch/.active and parse its .arch metadata.
set_active
Set the active libtorch variant by writing <root>/libtorch/.active.