neuralib.util.gpu

print_gpu_table

Print GPU info table and check the compatibility with backend package

gpu_available

Check if GPU is available.

check_mps_available

Check if metal is available

check_nvidia_cuda_available

Checks if the GPU driver reacts and otherwise raises a custom error.