Update scripts/utils.py

scripts/utils.py

@@ -24,7 +24,14 @@ from typing import List, Tuple
 from PIL import Image
 import trimesh
 
-providers = ['CUDAExecutionProvider', 'CPUExecutionProvider']
+providers = [
+    ('CUDAExecutionProvider', {
+        'device_id': 0,
+        'arena_extend_strategy': 'kSameAsRequested',
+        'gpu_mem_limit': 8 * 1024 * 1024 * 1024,
+        'cudnn_conv_algo_search': 'HEURISTIC',
+    })
+]
 
 session = new_session(providers=providers)