diff --git a/dockerfile/rocm5.7.x.dockerfile b/dockerfile/rocm5.7.x.dockerfile index 87af0e368..6599c0ca6 100644 --- a/dockerfile/rocm5.7.x.dockerfile +++ b/dockerfile/rocm5.7.x.dockerfile @@ -133,7 +133,7 @@ RUN cd /tmp && \ # Install RCCL RUN cd /opt/ && \ - git clone https://github.com/ROCmSoftwarePlatform/rccl.git && \ + git clone -b release/rocm-rel-5.7 https://github.com/ROCmSoftwarePlatform/rccl.git && \ cd rccl && \ mkdir build && \ cd build && \ diff --git a/dockerfile/rocm6.0.x.dockerfile b/dockerfile/rocm6.0.x.dockerfile index 648bc195d..bd33e289a 100644 --- a/dockerfile/rocm6.0.x.dockerfile +++ b/dockerfile/rocm6.0.x.dockerfile @@ -139,7 +139,7 @@ RUN cd /tmp && \ # Install RCCL RUN cd /opt/ && \ - git clone https://github.com/ROCmSoftwarePlatform/rccl.git && \ + git clone -b release/rocm-rel-6.0 https://github.com/ROCmSoftwarePlatform/rccl.git && \ cd rccl && \ mkdir build && \ cd build && \