Spaces:
Sleeping
Sleeping
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}") | |