neuralib.util.gpu

print_gpu_table	Print GPU info table and check the compatibility with backend package
gpu_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.

Get basic GPU info table and verbose

• cuda driver (Windows/Linux)

• metal backend support (MacOS)

from neuralib.util.gpu import print_gpu_table
print_gpu_table()