From 1d0489f91a1a39fed94c05d9aabae44663c62527 Mon Sep 17 00:00:00 2001
From: CarloLucibello <carlo.lucibello@gmail.com>
Date: Wed, 10 Jul 2024 09:42:05 +0200
Subject: [PATCH] functional only if cuDNN is functional

---
 ext/DeviceUtilsCUDAExt.jl | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/ext/DeviceUtilsCUDAExt.jl b/ext/DeviceUtilsCUDAExt.jl
index 6bdc768..22a332e 100644
--- a/ext/DeviceUtilsCUDAExt.jl
+++ b/ext/DeviceUtilsCUDAExt.jl
@@ -38,9 +38,12 @@ function _check_use_cuda!()
             @warn """
             cuDNN is not functional. Some functionality will not be available.
             """ maxlog=1
-        end
-    end
 
+            # We make the device selectable only if cuDNN is functional
+            # to avoid issues with convolutions and other deep learning operations
+            USE_CUDA_GPU[] = false
+        end        
+    end
     return
 end