From 48ddb5333208231f932710766ebd10175656acda Mon Sep 17 00:00:00 2001
From: Julien <182520+JulienMaille@users.noreply.github.com>
Date: Wed, 23 Sep 2020 11:15:02 +0200
Subject: [PATCH] Merge pull request #18386 from JulienMaille:patch-1

* Make sure there is a cuda device before getting it

* Update init.hpp
---
 modules/dnn/src/cuda4dnn/init.hpp | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/modules/dnn/src/cuda4dnn/init.hpp b/modules/dnn/src/cuda4dnn/init.hpp
index b548f0958a..e9d997311f 100644
--- 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;