1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20
#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; }