Merge pull request #18386 from JulienMaille:patch-1

* Make sure there is a cuda device before getting it * Update init.hpp
5 years ago · 48ddb53332
parent f584c6d723
commit 48ddb53332
1 changed files with 6 additions and 0 deletions
--- a/modules/dnn/src/cuda4dnn/init.hpp
+++ b/modules/dnn/src/cuda4dnn/init.hpp
@ -57,6 +57,9 @@ namespace cv { namespace dnn { namespace cuda4dnn {

    bool isDeviceCompatible()
    {
+        if (getDeviceCount() <= 0)
+            return false;
+
        int device_id = getDevice();
        if (device_id < 0)
            return false;
@ -77,6 +80,9 @@ namespace cv { namespace dnn { namespace cuda4dnn {

    bool doesDeviceSupportFP16()
    {
+        if (getDeviceCount() <= 0)
+            return false;
+
        int device_id = getDevice();
        if (device_id < 0)
            return false;