| #include <cuda_runtime.h> | |
| #include <iostream> | |
| int cuda_devices(void) { | |
| int result = 0; | |
| cudaGetDeviceCount(&result); | |
| return result; | |
| } | |
| int main(void) { | |
| int n = cuda_devices(); | |
| if (n == 0) { | |
| std::cout << "No CUDA hardware found. Exiting.\n"; | |
| return 0; | |
| } | |
| std::cout << "Found " << n << " CUDA devices.\n"; | |
| return 0; | |
| } |