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

tweaks

This commit is contained in:
Ian 2020-03-01 19:40:03 -05:00
parent 9b2f4919ee
commit 7555e488c6
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/Flux.jl
View File

@ -39,11 +39,10 @@ include("data/Data.jl")
include("deprecations.jl") include("deprecations.jl")
include(joinpath(@__DIR__, "cuda/cuda.jl")) include("cuda/cuda.jl")
function __init__() function __init__()
use_cuda[] = CuArrays.functional() # Can be overridden after load with Flux.use_cuda[] = false use_cuda[] = CuArrays.functional() # Can be overridden after load with `Flux.use_cuda[] = false`
if CuArrays.functional() if CuArrays.functional()
if !CuArrays.has_cudnn() if !CuArrays.has_cudnn()
@warn "CuArrays.jl found cuda, but did not find libcudnn. Some functionality will not be available." @warn "CuArrays.jl found cuda, but did not find libcudnn. Some functionality will not be available."