From 7555e488c626e45b9e0b4d91e56ecb6775384b06 Mon Sep 17 00:00:00 2001 From: Ian Date: Sun, 1 Mar 2020 19:40:03 -0500 Subject: [PATCH] tweaks --- src/Flux.jl | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/src/Flux.jl b/src/Flux.jl index 8fb0f406..5afa1fc0 100644 --- a/src/Flux.jl +++ b/src/Flux.jl @@ -39,11 +39,10 @@ include("data/Data.jl") include("deprecations.jl") -include(joinpath(@__DIR__, "cuda/cuda.jl")) - +include("cuda/cuda.jl") 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.has_cudnn() @warn "CuArrays.jl found cuda, but did not find libcudnn. Some functionality will not be available."