neuralib.util.gpu.print_gpu_table

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

Print GPU info table and check the compatibility with backend package

Parameters:

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

Returns:

Return type:

None