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 \