Relax dependency on the toolkit version.

Author: maleadt

CUDACore/src/CUDACore.jl

@@ -35,11 +35,6 @@ using CUDA_Compiler_jll
 
 import CUDA_Runtime_jll
 const local_toolkit = CUDA_Runtime_jll.host_platform["cuda_local"] == "true"
-const toolkit_version = if CUDA_Runtime_jll.host_platform["cuda"] == "none"
-    nothing
-else
-    parse(VersionNumber, CUDA_Runtime_jll.host_platform["cuda"])
-end
 if local_toolkit
     using CUDA_Runtime_Discovery
     const CUDA_Runtime = CUDA_Runtime_Discovery

CUDACore/src/initialization.jl

@@ -157,19 +157,21 @@ function __init__()
         @error "This version of CUDA.jl only supports CUDA 12 or higher (your toolkit provides CUDA $runtime)"
     end
 
-    # ensure the loaded runtime matches what we precompiled for.
-    if toolkit_version == nothing
-        @error """CUDA.jl was precompiled without knowing the CUDA toolkit version. This is unsupported.
-                  You should either precompile CUDA.jl in an environment where the CUDA toolkit is available,
-                  or call `CUDA.set_runtime_version!` to specify which CUDA version to use."""
-    elseif Base.thisminor(runtime) != Base.thisminor(toolkit_version)
-        # this can only happen with a local toolkit, but let's always check to be sure
-        if local_toolkit
-            @error """You are using a local CUDA $(Base.thisminor(runtime)) toolkit, but CUDA.jl was precompiled for CUDA $(Base.thisminor(toolkit_version)). This is unsupported.
-                      Call `CUDA.set_runtime_version!` to update the CUDA version to match your local installation."""
-        else
-            @error """You are using CUDA $(Base.thisminor(runtime)), but CUDA.jl was precompiled for CUDA $(Base.thisminor(toolkit_version)).
+    # ensure the loaded runtime matches the artifact we were compiled against
+    if !local_toolkit
+        if CUDA_Runtime_jll.host_platform["cuda"] == "none"
+            @error """CUDA.jl was precompiled without knowing the CUDA toolkit version. This is unsupported.
+                      You should either precompile CUDA.jl in an environment where the CUDA driver is available,
+                      or call `CUDA.set_runtime_version!` to specify which CUDA version to use at run time."""
+            _initialization_error[] = "Precompiled without CUDA toolkit version"
+            return
+        end
+
+        artifact = parse(VersionNumber, CUDA_Runtime_jll.host_platform["cuda"])
+        if Base.thisminor(runtime) != Base.thisminor(artifact)
+            @error """You are using CUDA $runtime, but CUDA.jl was precompiled for CUDA $artifact.
                       This is unexpected; please file an issue."""
+            _initialization_error[] = "CUDA version mismatch"
         end
     end
 

LocalPreferences.toml

@@ -3,8 +3,7 @@
 # set this preference if you want to precompile CUDA.jl in an envirnonment without CUDA
 #version = "11.8"
 
-# whether to use a local CUDA installation. if CUDA isn't available during precompilation,
-# you will also need to set the "version" preference, matching the local CUDA version.
+# whether to use a local CUDA installation.
 #local = "false"
 
 [CUDA]