neuralib.util.gpu.gpu_available

neuralib.util.gpu.gpu_available(backend, *, check_smi=False)[source]

Check if GPU is available.

Parameters:

backend (Literal['tensorflow', 'torch']) – {‘torch’, ‘tensorflow’}
check_smi (bool) – check if nvidia-smi is runnable

Returns:

Return type:

bool