cuda: add driver version to minimal.cu

danbev · Oct 10, 2024 · 6869757 · 6869757
1 parent 15f0a5c
commit 6869757
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/gpu/cuda/src/minimal.cu b/gpu/cuda/src/minimal.cu
@@ -9,15 +9,20 @@ int main() {
 
     int runtimeVersion = 0;
 
-    // Get CUDA runtime version
     err = cudaRuntimeGetVersion(&runtimeVersion);
     if (err != cudaSuccess) {
         printf("cudaRuntimeGetVersion failed: %s\n", cudaGetErrorString(err));
         return 1;
     }
-
     printf("CUDA Runtime version: %d.%d\n", runtimeVersion / 1000, (runtimeVersion % 100) / 10);
 
+    int driverVersion = 0;
+    err = cudaDriverGetVersion(&driverVersion);
+    if (err != cudaSuccess) {
+        printf("cudaDriverGetVersion failed: %s\n", cudaGetErrorString(err));
+        return 1;
+    }
+    printf("CUDA Driver version: %d.%d\n", driverVersion / 1000, (driverVersion % 100) / 10);
 
     err = cudaGetDeviceCount(&count);
     if (err != cudaSuccess) {