Explorar el Código

feat:添加ninja-build

zhaohaipeng hace 2 meses
padre
commit
66d71308bf
Se han modificado 1 ficheros con 2 adiciones y 0 borrados
  1. 2 0
      compose.base.yml

+ 2 - 0
compose.base.yml

@@ -14,6 +14,8 @@ services:
     environment:
       COMPILE: ${COMPILE:-0}
       PYTORCH_CUDA_ALLOC_CONF: ${PYTORCH_CUDA_ALLOC_CONF:-expandable_segments:True}
+      TORCH_LOGS: +dynamo
+      TORCHDYNAMO_VERBOSE: 1
     # GPU (remove this block if CPU-only):
     deploy:
       resources: