pkgs/by-name/bl/blender/test-cuda.py at devShellTools-shell

tjh.dev / nixpkgs

fork atom

nixpkgs mirror (for testing) github.com/NixOS/nixpkgs

nix

fork atom

nixpkgs / pkgs / by-name / bl / blender / test-cuda.py

at devShellTools-shell 8 lines 255 B view raw

wrap content

1import bpy
2
3preferences = bpy.context.preferences.addons["cycles"].preferences
4devices = preferences.get_devices_for_type("CUDA")
5ids = [d.id for d in devices]
6
7assert any("CUDA" in i for i in ids), f"CUDA not present in {ids}"
8print("CUDA is available")