I was trying to run the Flux modelzoo MNIST mlp.jl example and I notice these warning coming from CuArrays.jl:
julia> include("mlp.jl") [ Info: Precompiling Flux [587475ba-b771-5e3f-ad9e-33799f191a9c] ┌ Warning: CuArrays.jl only supports CUDNN v7.6 or higher └ @ CuArrays ~/.julia/packages/CuArrays/A6GUx/src/CuArrays.jl:122 ┌ Warning: You are using CUDNN 7.1.4 for CUDA 9.0.0 with CUDA toolkit 9.2.148; these might be incompatible.
However, I’ve got CUDNN 7.6.5 installed and nvidia-smi reports that I’m on CUDA Version 10.1.
Why does CuArrays.jl think I’m using CUDA 9.0.0 and CUDNN 7.1.4?
Is there some configuration to tell CuArrays.jl to use different versions of CUDA/CUDNN libs?
I also get the same warnings above if I just do:
pkg> test CuArrays
Julia 1.3.1, CuArrays v1.7.3