cuda_driver_api_init_device_info_context_management.py

1from cuda import cuda, cudart
2
3# Initialize the Driver API
4err, = cuda.cuInit(0)
5if err != cuda.CUresult.CUDA_SUCCESS:
6    raise RuntimeError("CUDA initialization failed")
7
8# Get the first device
9err, device = cuda.cuDeviceGet(0)
10if err != cuda.CUresult.CUDA_SUCCESS:
11    raise RuntimeError("Failed to get CUDA device")
12
13# Create a context
14err, context = cuda.cuCtxCreate(0, device)
15if err != cuda.CUresult.CUDA_SUCCESS:
16    raise RuntimeError("Failed to create CUDA context")
17
18# Get device name
19err, name = cuda.cuDeviceGetName(100, device)
20if err == cuda.CUresult.CUDA_SUCCESS:
21    print(f"Device Name: {name.decode('utf-8')}")
22
23# Clean up
24cuda.cuCtxDestroy(context)