revert libnccl

Author: sirutBuasai

tensorflow/training/docker/2.18/py3/cu125/Dockerfile.gpu

@@ -51,7 +51,7 @@ ARG BRANCH_OFI=1.13.2-aws
 ARG CUDA=12.5
 ARG CUDA_DASH=12-5
 ARG CUDNN=9.3.0.75
-ARG NCCL_VERSION=2.23.4
+ARG NCCL_VERSION=2.23.4-1+cuda12.5
 
 
 # To be passed to ec2 and sagemaker stages
@@ -106,8 +106,8 @@ RUN apt-get update && apt-get install -y --no-install-recommends --allow-unauthe
  && apt-get install -y --no-install-recommends --allow-unauthenticated --allow-change-held-packages \
    libcublas-dev-${CUDA_DASH} \
    libcublas-${CUDA_DASH} \
-   libnccl2=${NCCL_VERSION} \
-   libnccl-dev=${NCCL_VERSION} \
+   libnccl2 \
+   libnccl-dev \
  && rm -rf /var/lib/apt/lists/* \
  && apt-get clean \
  && mkdir -p /var/run/sshd