blob: ed9333efb6e818149ec5fbea704cc99f4a7d4957 [file] [log] [blame]
#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;
}