Spaces:
Sleeping
Sleeping
File size: 463 Bytes
8d4d3e5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
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}")
|