added CUDA version info to gpu tests

This commit is contained in:
Vladislav Vinogradov
2011-06-29 12:00:00 +00:00
parent 52bdfbc7fc
commit 724a3c1473
2 changed files with 4 additions and 1 deletions

View File

@@ -46,7 +46,7 @@
void print_info() void print_info()
{ {
#if defined _WIN32 #if defined _WIN32
# if define _WIN64 # if defined _WIN64
puts("OS: Windows 64\n"); puts("OS: Windows 64\n");
# else # else
puts("OS: Windows 32\n"); puts("OS: Windows 32\n");
@@ -65,6 +65,8 @@ void print_info()
# endif # endif
#endif #endif
printf("CUDA version: %d\n\n", CUDART_VERSION);
int deviceCount = cv::gpu::getCudaEnabledDeviceCount(); int deviceCount = cv::gpu::getCudaEnabledDeviceCount();

View File

@@ -50,6 +50,7 @@
#include <string> #include <string>
#include <algorithm> #include <algorithm>
#include <iterator> #include <iterator>
#include <cuda_runtime_api.h>
#include "cvconfig.h" #include "cvconfig.h"
#include "opencv2/core/core.hpp" #include "opencv2/core/core.hpp"
#include "opencv2/highgui/highgui.hpp" #include "opencv2/highgui/highgui.hpp"