Use correct versions for library discovery.

JuliaGPU · Feb 11, 2020 · a0367f4 · a0367f4
1 parent 93493bc
commit a0367f4
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/src/CuArrays.jl b/src/CuArrays.jl
@@ -67,8 +67,13 @@ function __init__()
 
     try
         # discover libraries
-        toolkit = find_toolkit()
-        for name in ("cublas", "cusparse", "cusolver", "cufft", "curand", "cudnn", "cutensor")
+        for (name,version) in  (("cublas", CUDAnative.version()),
+                                ("cusparse", CUDAnative.version()),
+                                ("cusolver", CUDAnative.version()),
+                                ("cufft", CUDAnative.version()),
+                                ("curand", CUDAnative.version()),
+                                ("cudnn", v"7"),
+                                ("cutensor", v"1"))
             mod = getfield(CuArrays, Symbol(uppercase(name)))
             lib = Symbol("lib$name")
             handle = getfield(mod, lib)
@@ -82,7 +87,7 @@ function __init__()
 
             # check if we can't find the library
             if Libdl.dlopen_e(handle[]) == C_NULL
-                path = find_cuda_library(name, CUDAnative.prefix(), [CUDAnative.version()])
+                path = find_cuda_library(name, CUDAnative.prefix(), [version])
                 if path !== nothing
                     handle[] = path
                 end