mirror of
https://github.com/opencv/opencv.git
synced 2024-11-30 06:10:02 +08:00
71 lines
2.9 KiB
C++
71 lines
2.9 KiB
C++
#include "perf_precomp.hpp"
|
|
|
|
static void printOsInfo()
|
|
{
|
|
#if defined _WIN32
|
|
# if defined _WIN64
|
|
printf("[----------]\n[ GPU INFO ] \tRun on OS Windows x64.\n[----------]\n"), fflush(stdout);
|
|
# else
|
|
printf("[----------]\n[ GPU INFO ] \tRun on OS Windows x32.\n[----------]\n"), fflush(stdout);
|
|
# endif
|
|
#elif defined linux
|
|
# if defined _LP64
|
|
printf("[----------]\n[ GPU INFO ] \tRun on OS Linux x64.\n[----------]\n"), fflush(stdout);
|
|
# else
|
|
printf("[----------]\n[ GPU INFO ] \tRun on OS Linux x32.\n[----------]\n"), fflush(stdout);
|
|
# endif
|
|
#elif defined __APPLE__
|
|
# if defined _LP64
|
|
printf("[----------]\n[ GPU INFO ] \tRun on OS Apple x64.\n[----------]\n"), fflush(stdout);
|
|
# else
|
|
printf("[----------]\n[ GPU INFO ] \tRun on OS Apple x32.\n[----------]\n"), fflush(stdout);
|
|
# endif
|
|
#endif
|
|
|
|
}
|
|
|
|
static void printCudaInfo()
|
|
{
|
|
printOsInfo();
|
|
#ifndef HAVE_CUDA
|
|
printf("[----------]\n[ GPU INFO ] \tOpenCV was built without CUDA support.\n[----------]\n"), fflush(stdout);
|
|
#else
|
|
int driver;
|
|
cudaDriverGetVersion(&driver);
|
|
|
|
printf("[----------]\n"), fflush(stdout);
|
|
printf("[ GPU INFO ] \tCUDA Driver version: %d.\n", driver), fflush(stdout);
|
|
printf("[ GPU INFO ] \tCUDA Runtime version: %d.\n", CUDART_VERSION), fflush(stdout);
|
|
printf("[----------]\n"), fflush(stdout);
|
|
|
|
printf("[----------]\n"), fflush(stdout);
|
|
printf("[ GPU INFO ] \tGPU module was compiled for the following GPU archs.\n"), fflush(stdout);
|
|
printf("[ BIN ] \t%s.\n", CUDA_ARCH_BIN), fflush(stdout);
|
|
printf("[ PTX ] \t%s.\n", CUDA_ARCH_PTX), fflush(stdout);
|
|
printf("[----------]\n"), fflush(stdout);
|
|
|
|
printf("[----------]\n"), fflush(stdout);
|
|
int deviceCount = cv::gpu::getCudaEnabledDeviceCount();
|
|
printf("[ GPU INFO ] \tCUDA device count:: %d.\n", deviceCount), fflush(stdout);
|
|
printf("[----------]\n"), fflush(stdout);
|
|
|
|
for (int i = 0; i < deviceCount; ++i)
|
|
{
|
|
cv::gpu::DeviceInfo info(i);
|
|
|
|
printf("[----------]\n"), fflush(stdout);
|
|
printf("[ DEVICE ] \t# %d %s.\n", i, info.name().c_str()), fflush(stdout);
|
|
printf("[ ] \tCompute capability: %d.%d\n", (int)info.majorVersion(), (int)info.minorVersion()), fflush(stdout);
|
|
printf("[ ] \tMulti Processor Count: %d\n", info.multiProcessorCount()), fflush(stdout);
|
|
printf("[ ] \tTotal memory: %d Mb\n", static_cast<int>(static_cast<int>(info.totalMemory() / 1024.0) / 1024.0)), fflush(stdout);
|
|
printf("[ ] \tFree memory: %d Mb\n", static_cast<int>(static_cast<int>(info.freeMemory() / 1024.0) / 1024.0)), fflush(stdout);
|
|
if (!info.isCompatible())
|
|
printf("[ GPU INFO ] \tThis device is NOT compatible with current GPU module build\n");
|
|
printf("[----------]\n"), fflush(stdout);
|
|
}
|
|
|
|
#endif
|
|
}
|
|
|
|
CV_PERF_TEST_MAIN(gpu, printCudaInfo())
|