about summary refs log tree commit diff
path: root/pkgs/applications/misc/blender/test-cuda.py
diff options
context:
space:
mode:
Diffstat (limited to 'pkgs/applications/misc/blender/test-cuda.py')
-rw-r--r--pkgs/applications/misc/blender/test-cuda.py8
1 files changed, 8 insertions, 0 deletions
diff --git a/pkgs/applications/misc/blender/test-cuda.py b/pkgs/applications/misc/blender/test-cuda.py
new file mode 100644
index 0000000000000..8a3ec57347592
--- /dev/null
+++ b/pkgs/applications/misc/blender/test-cuda.py
@@ -0,0 +1,8 @@
+import bpy
+
+preferences = bpy.context.preferences.addons["cycles"].preferences
+devices = preferences.get_devices_for_type("CUDA")
+ids = [d.id for d in devices]
+
+assert any("CUDA" in i for i in ids), f"CUDA not present in {ids}"
+print("CUDA is available")