diff --git a/dockerfile/rocm5.1.1-pytorch1.10.0.dockerfile b/dockerfile/rocm5.1.1-pytorch1.10.0.dockerfile index 8c7bc88e..ebd83bf3 100644 --- a/dockerfile/rocm5.1.1-pytorch1.10.0.dockerfile +++ b/dockerfile/rocm5.1.1-pytorch1.10.0.dockerfile @@ -110,13 +110,13 @@ RUN cd /tmp && \ cd /tmp && \ rm -rf mlc -# Install rccl with commitid 700b473 +# Install rccl with commitid 6707a27 RUN cd /tmp && \ git clone https://github.com/ROCmSoftwarePlatform/rccl.git && \ - cd rccl && git checkout 700b473 && \ + cd rccl && git checkout 6707a27 && \ mkdir build && cd build && \ - CXX=/opt/rocm/bin/hipcc cmake -DCMAKE_INSTALL_PREFIX=/opt/rocm/rccl .. && \ - make -j ${NUM_MAKE_JOBS} && make install && \ + CXX=/opt/rocm/bin/hipcc cmake -DCMAKE_INSTALL_PREFIX=/usr/local .. && \ + make && make install && \ cd /tmp && \ rm -rf rccl