blob: 56688303f8336f9da2b943b6b303f1023bec73bd (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
|
#include <cuda_runtime.h>
#include <iostream>
int cuda_devices(void) {
int result = 0;
cudaGetDeviceCount(&result);
return result;
}
int main(void) {
std::cout << "Compiled against CUDA version: " << CUDART_VERSION << "\n";
int runtime_version = 0;
cudaError_t r = cudaRuntimeGetVersion(&runtime_version);
if (r != cudaSuccess) {
std::cout << "Couldn't obtain CUDA runtime version (error " << r << "). Exiting.\n";
return -1;
}
std::cout << "CUDA runtime version: " << runtime_version << "\n";
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;
}
|