main.cu 364 B

123456789101112131415161718192021
  1. #include <cuda_runtime.h>
  2. #include <iostream>
  3. int cuda_devices(void) {
  4. int result = 0;
  5. cudaGetDeviceCount(&result);
  6. return result;
  7. }
  8. int main(void) {
  9. int n = cuda_devices();
  10. if (n == 0) {
  11. std::cout << "No Cuda hardware found. Exiting.\n";
  12. return 0;
  13. }
  14. std::cout << "Found " << n << "Cuda devices.\n";
  15. return 0;
  16. }