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

Don't depend on unreleased CuArrays.

This commit is contained in:
Tim Besard 2019-08-27 09:40:22 +02:00
parent 6ad3cdd138
commit 4fef9d8508
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
Project.toml
View File

@ -25,10 +25,11 @@ Tracker = "9f7883ad-71c0-57eb-9f7f-b5c9e6d3789c"
ZipFile = "a5390f91-8eb1-5f08-bee0-b1d1ffed6cea"
[compat]
CUDAapi = "1.1"
CuArrays = "1.2"
NNlib = "0.6"
Tracker = "0.2"
julia = "0.7, 1"
CuArrays = "1.3"
[extras]
Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"

2
src/cuda/cuda.jl
View File

@ -2,7 +2,7 @@ module CUDA
using ..CuArrays
if has_cudnn()
if CuArrays.libcudnn !== nothing # TODO: use CuArrays.has_cudnn()
include("curnn.jl")
include("cudnn.jl")
else