Max gpu mem

xuhao1 · Nov 18, 2022 · 7d22570 · 7d22570
1 parent b84c93a
commit 7d22570
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/taichislam_node.py b/scripts/taichislam_node.py
@@ -30,7 +30,7 @@ class TaichiSLAMNode:
     def __init__(self):
         self.init_params()
         if self.cuda:
-            ti.init(arch=ti.cuda, dynamic_index=True, offline_cache=True, packed=False, debug=False)
+            ti.init(arch=ti.cuda, dynamic_index=True, offline_cache=True, packed=True, debug=False, device_memory_GB=1.5)
         else:
             ti.init(arch=ti.cpu, dynamic_index=True, offline_cache=True, packed=True)
         self.disp_level = 0