neuralib.util.gpu.check_mps_available

neuralib.util.gpu.check_mps_available(backend)[source]

Check if metal is available

Parameters:

backend (Literal['tensorflow', 'torch']) – {‘torch’, ‘tensorflow’}

Returns:

bool

Return type:

bool