This module contains helper functions that check if various environmental factors are available

is_tpu_available[source]

is_tpu_available(check_device=True)

Checks if torch_xla is installed and potentially if a TPU is in the environment

is_bf16_available[source]

is_bf16_available(ignore_tpu=False)

Checks if bf16 is supported, optionally ignoring the TPU

is_multigpu_available[source]

is_multigpu_available()

Checks if number of cuda devices available > 1