diff --git a/src/cuda/cuda.jl b/src/cuda/cuda.jl index fe36bf5d..15126aca 100644 --- a/src/cuda/cuda.jl +++ b/src/cuda/cuda.jl @@ -2,6 +2,6 @@ module CUDA using ..CuArrays -CuArrays.cudnn_available() && include("cudnn.jl") +CuArrays.libcudnn != nothing && include("cudnn.jl") end diff --git a/test/cuda/cuda.jl b/test/cuda/cuda.jl index 16f90e89..1f54d1b9 100644 --- a/test/cuda/cuda.jl +++ b/test/cuda/cuda.jl @@ -36,4 +36,4 @@ Flux.back!(sum(l)) end -CuArrays.cudnn_available() && include("cudnn.jl") +CuArrays.libcudnn != nothing && include("cudnn.jl")