functional only if cuDNN is functional

LuxDL · Jul 10, 2024 · 1d0489f · 1d0489f
1 parent a12e4eb
commit 1d0489f
Showing 1 changed file with 5 additions and 2 deletions.
diff --git 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