From c4142cc566c4fa34b422cd245c88d327b516cd30 Mon Sep 17 00:00:00 2001
From: hailin <hailin@gdzx.xyz>
Date: Wed, 2 Jul 2025 16:01:23 +0800
Subject: [PATCH] .

---
 Dockerfile | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/Dockerfile b/Dockerfile
index 8e835d8..cf5009c 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -4,7 +4,14 @@
 ARG CUDA_VERSION=12.6.1
 FROM nvidia/cuda:${CUDA_VERSION}-devel-ubuntu22.04 AS builder-torch
 
-ENV USE_DISTRIBUTED=1
+ENV USE_CUDA=1 \
+    USE_DISTRIBUTED=1 \
+    USE_MPI=1 \
+    USE_GLOO=1 \
+    USE_NCCL=1 \
+    USE_SYSTEM_NCCL=1 \
+    BUILD_TEST=0
+    
 ARG MAX_JOBS=90                       
 ENV DEBIAN_FRONTEND=noninteractive \
     PYTHONUNBUFFERED=1 LANG=C.UTF-8 LC_ALL=C.UTF-8 \