radames commited on
Commit
9108d7c
1 Parent(s): 4373daa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +21 -0
app.py CHANGED
@@ -6,6 +6,27 @@ import tempfile
6
  import molecularnodes as mn
7
  import os
8
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
9
 
10
  window = bpy.context.window
11
  screen = window.screen
 
6
  import molecularnodes as mn
7
  import os
8
 
9
+ def enable_GPUS():
10
+ bpy.data.scenes[0].render.engine = "CYCLES" #"CYCLES"
11
+ # Set the device_type
12
+ bpy.context.preferences.addons[
13
+ "cycles"
14
+ ].preferences.compute_device_type = "CUDA" # or "OPENCL"
15
+
16
+ # Set the device and feature set
17
+ bpy.context.scene.cycles.device = "GPU"
18
+
19
+ for scene in bpy.data.scenes:
20
+ scene.cycles.device = "GPU"
21
+
22
+ bpy.context.preferences.addons["cycles"].preferences.get_devices()
23
+ print(bpy.context.preferences.addons["cycles"].preferences.compute_device_type)
24
+ for d in bpy.context.preferences.addons["cycles"].preferences.devices:
25
+ d["use"] = True # Using all devices, include GPU and CPU
26
+ print(d["name"])
27
+
28
+
29
+ enable_GPUS()
30
 
31
  window = bpy.context.window
32
  screen = window.screen