mirror of https://github.com/opencv/opencv.git
Open Source Computer Vision Library
https://opencv.org/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
25 lines
723 B
25 lines
723 B
#include <iostream> |
|
#include <sstream> |
|
#include <list> |
|
|
|
int main() |
|
{ |
|
std::ostringstream arch; |
|
std::list<std::string> archs; |
|
|
|
int count = 0; |
|
if (cudaSuccess != cudaGetDeviceCount(&count)){ return -1; } |
|
if (count == 0) { return -1; } |
|
for (int device = 0; device < count; ++device) |
|
{ |
|
cudaDeviceProp prop; |
|
if (cudaSuccess != cudaGetDeviceProperties(&prop, device)){ continue; } |
|
arch << prop.major << "." << prop.minor; |
|
archs.push_back(arch.str()); |
|
arch.str(""); |
|
} |
|
archs.unique(); #Some devices might have the same arch |
|
for (std::list<std::string>::iterator it=archs.begin(); it!=archs.end(); ++it) |
|
std::cout << *it << " "; |
|
return 0; |
|
}
|
|
|