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