is_torch_compile_compatible

tiatoolbox . models . architecture . utils . is_torch_compile_compatible

is_torch_compile_compatible()

Check if the current GPU is compatible with torch-compile.

Returns:

True if current GPU is compatible with torch-compile, False otherwise.

Raises:

Warning if GPU is not compatible with torch.compile. –

Return type:

bool