1 #include <iostream> 2 3 #include <cuda.h> 4 use_cuda(void)5extern "C" int use_cuda(void) 6 { 7 int nDevices = 0; 8 cudaError_t err = cudaGetDeviceCount(&nDevices); 9 if (err != cudaSuccess) { 10 std::cerr << "Failed to retrieve the number of CUDA enabled devices" 11 << std::endl; 12 return 1; 13 } 14 std::cout << "Found " << nDevices << " CUDA enabled devices" << std::endl; 15 return 0; 16 } 17