diff --git a/Dockerfile b/Dockerfile
index 6ac4319..15c7889 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -10,11 +10,12 @@ ENV DEBIAN_FRONTEND=noninteractive \
     USE_CUDA=1 USE_DISTRIBUTED=0 BUILD_TEST=0 TORCH_CUDA_ARCH_LIST="7.5;8.0;8.6;9.0"
 
 RUN apt-get update && apt-get install -y --no-install-recommends \
-        python3 python3-pip python3-distutils \
-        git cmake ninja-build \
-        libopenblas-dev libopenmpi-dev libnccl-dev \
-        libjpeg-dev libpng-dev ca-certificates && \
-    python3 -m pip install --no-cache-dir --upgrade pip wheel setuptools sympy
+      python3 python3-pip python3-distutils git cmake ninja-build \
+      libopenblas-dev libopenmpi-dev \
+      libnccl2=2.22.3-1+cuda12.6 \
+      libnccl-dev=2.22.3-1+cuda12.6 \
+      libjpeg-dev libpng-dev ca-certificates && \
+    python3 -m pip install --no-cache-dir --upgrade pip wheel setuptoo
 
 WORKDIR /opt
 RUN git clone --recursive -b v2.7.1 https://github.com/pytorch/pytorch.git
@@ -32,9 +33,11 @@ FROM nvidia/cuda:${CUDA_VERSION}-devel-ubuntu22.04 AS builder-extras
 ENV DEBIAN_FRONTEND=noninteractive PYTHONUNBUFFERED=1 LANG=C.UTF-8 LC_ALL=C.UTF-8
 
 RUN apt-get update && apt-get install -y --no-install-recommends \
-        python3 python3-pip python3-distutils \
-        git build-essential cmake ninja-build \
-        libjpeg-dev libpng-dev ca-certificates && \
+      python3 python3-pip python3-distutils git build-essential \
+      cmake ninja-build libjpeg-dev libpng-dev ca-certificates \
+      libopenmpi-dev \
+      libnccl2=2.22.3-1+cuda12.6 \
+      libnccl-dev=2.22.3-1+cuda12.6 && \
     python3 -m pip install --no-cache-dir --upgrade pip wheel setuptools
 
 # ── 安装自编 torch 轮子 ──────────────────────────────────────────────────────