improve docstring

LuxDL · Jul 10, 2024 · 336e06e · 336e06e
1 parent 8c1028a
commit 336e06e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ext/DeviceUtilsCUDAExt.jl b/ext/DeviceUtilsCUDAExt.jl
@@ -21,7 +21,7 @@ function _check_use_cuda!()
         if !cudnn_loaded
             @warn """
             cuDNN is not loaded. Some functionality will not be available.
-            Load cuDNN by running `using cuDNN` or `using LuxCUDA`.
+            Load cuDNN by running `using CUDA, cuDNN` or `using LuxCUDA`.
             """ maxlog=1
         else
             cuDNN = Base.loaded_modules[cudnnid]