import pycuda.driver as cuda
import pycuda.autoinit

try:
    cuda.init()
    print(f"Detected {cuda.Device.count()} CUDA device(s).")
    
    for i in range(cuda.Device.count()):
        gpu = cuda.Device(i)
        print(f"Device {i}: {gpu.name()}")
        print(f"  Compute Capability: {gpu.compute_capability()}")
        print(f"  Total Memory: {gpu.total_memory() // (1024 ** 2)} MB")
except cuda.Error as e:
    print(f"CUDA initialization failed: {e}")