Add function to detect nvcc version.

README.md

@@ -18,6 +18,10 @@ Compiles CUDA code using the Visual Studio CUDA Toolkit extension on Windows and
 - `cudaGenLineInfo` (Bool) -> generates line info.
 - `cudaIntDir` (String) -> Intermediary directory for CUDA files - Windows only.
 
+The following functions are provided:
+- `detectNvccVersion()` -> try to detect the default version of nvcc on the system.
+- `detectNvccVersion(cudaPath)` -> try to detect the version of nvcc from a provided path.
+
 ----------------
 Notes for Windows:
 ----------------

premake5-cuda.lua

@@ -5,3 +5,5 @@ if os.target() == "windows" then
 elseif os.target() == "linux" then
     dofile("src/premake5-cuda-nvcc.lua")
 end
+
+dofile("src/utils.lua")

src/utils.lua

@@ -0,0 +1,24 @@
+-- * Returns the version of nvcc detected on the system, or 0 if undetected.
+function detectNvccVersion(cudaPath)
+  local errorCode = 1
+  local output
+
+  if cudaPath ~= nil then
+        output, errorCode = os.outputof("/usr/local/cuda/bin/nvcc --version")
+  else
+    if os.host() == "linux" then
+      output, errorCode = os.outputof("/usr/local/cuda/bin/nvcc --version")
+    elseif os.host() == "windows" then
+      output, errorCode = os.outputof("nvcc --version")
+    else
+      return 0
+    end
+  end
+
+  if not errorCode == 0 then
+    return 0
+  end
+
+  local versionString = output:match("cuda_(%S+)")
+  return versionString:match("[^.]*.[^.]*")
+end

test/premake5.lua

@@ -1,5 +1,7 @@
 require("../premake5-cuda")
 
+local cudaVer = detectNvccVersion()
+
 workspace "ExampleProject"
 
   language "C++"
@@ -48,7 +50,7 @@ project "ExampleProjectExe"
 
   files { "exe/cpp/**.cpp" }
 
-  buildcustomizations "BuildCustomizations/CUDA 12.1"
+  buildcustomizations("BuildCustomizations/CUDA " .. cudaVer)
 
   externalwarnings "Off" -- thrust gives a lot of warnings
 
@@ -82,7 +84,7 @@ project "ExampleProjectDLL"
 
   location "out/%{prj.name}"
 
-  buildcustomizations "BuildCustomizations/CUDA 12.1"
+  buildcustomizations("BuildCustomizations/CUDA " .. cudaVer)
 
   if os.target() == "windows" then
     -- Just in case we want the VS CUDA extension to use a custom version of CUDA
@@ -100,7 +102,7 @@ project "ExampleProjectDLL"
   defines { "PREMAKE_CUDA_EXPORT_API" }
 
   -- Let's compile for all supported architectures (and also in parallel with -t0)
-  cudaCompilerOptions {"-arch=all", "-t0"} 
+  cudaCompilerOptions {"-arch=all", "-t0"}
 
   filter "configurations:debug"
     cudaLinkerOptions { "-g" }