TastyPancakes
/
Flux.jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

libcudnn check fix

This commit is contained in:
Mike J Innes 2019-08-27 15:41:23 +01:00
parent 9cd97f06f7
commit 61a8cfd6ee
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
test/cuda/cuda.jl
View File

@ -48,7 +48,7 @@ end
@test y[3,:] isa CuArray
end
if has_cudnn() != nothing
if CuArrays.libcudnn != nothing
@info "Testing Flux/CUDNN"
include("cudnn.jl")
if !haskey(ENV, "CI_DISABLE_CURNN_TEST")