| #include <cuda_runtime.h> | |
| #include <stdio.h> | |
| int cuda_devices(void) { | |
| int result = 0; | |
| cudaGetDeviceCount(&result); | |
| return result; | |
| } | |
| int main(void) { | |
| int n = cuda_devices(); | |
| if (n == 0) { | |
| printf("No CUDA hardware found. Exiting.\n"); | |
| return 0; | |
| } | |
| printf("Found %i CUDA devices.\n", n); | |
| return 0; | |
| } |