Skip to content
voiage

voiage.core.gpu_acceleration.get_gpu_backend

get_gpu_backend

Section titled “get_gpu_backend”
get_gpu_backend() -> str

Get the available GPU backend.

Returns

Section titled “Returns”

str: Name of the available GPU backend (‘jax’, ‘cupy’, ‘torch’, or ‘none’)

Returns: str