Get CUDA properties

pytorch · Nov 20, 2024 · 2419d49 · 2419d49
1 parent fa0109a
commit 2419d49
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/benchmarks/decoders/benchmark_readme_data.json b/benchmarks/decoders/benchmark_readme_data.json
@@ -392,6 +392,7 @@
       "is_cuda_available": "True",
       "machine": "x86_64",
       "python_version": "3.12.5",
-      "system": "Linux"
+      "system": "Linux",
+      "cuda": "bogus"
   }
 }
diff --git a/benchmarks/decoders/generate_readme_data.py b/benchmarks/decoders/generate_readme_data.py
@@ -90,7 +90,11 @@ def main() -> None:
             "system": platform.system(),
             "machine": platform.machine(),
             "python_version": str(platform.python_version()),
-            "is_cuda_available": str(torch.cuda.is_available()),
+            "cuda": (
+                str(torch.cuda.get_device_properties(0))
+                if torch.cuda.is_available()
+                else "not available"
+            ),
         },
     }