Commit 724a3c14 authored by Vladislav Vinogradov's avatar Vladislav Vinogradov

added CUDA version info to gpu tests

parent 52bdfbc7
...@@ -46,7 +46,7 @@ ...@@ -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() ...@@ -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();
......
...@@ -50,6 +50,7 @@ ...@@ -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"
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment