Merge pull request #18386 from JulienMaille:patch-1

* Make sure there is a cuda device before getting it * Update init.hpp
2020-09-23 11:15:02 +02:00
parent f584c6d723
commit 48ddb53332
1 changed files with 6 additions and 0 deletions
@@ -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;