Skip to content

Commit

Permalink
Dockerfile: Update rccl version and fix issue in rocm5.1.1 dockerfile (
Browse files Browse the repository at this point in the history
…#354)

**Description**
Update rccl version and fix issue in rocm5.1.1 dockerfile.
  • Loading branch information
yukirora committed May 27, 2022
1 parent 81a4146 commit e08b6d3
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions dockerfile/rocm5.1.1-pytorch1.10.0.dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down

0 comments on commit e08b6d3

Please sign in to comment.