Skip to content

Commit

Permalink
install rccl to /usr/local
Browse files Browse the repository at this point in the history
  • Loading branch information
yukirora committed May 26, 2022
1 parent ef15413 commit 496f4e2
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion dockerfile/rocm5.1.1-pytorch1.10.0.dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -116,7 +116,7 @@ RUN cd /tmp && \
git clone https://github.com/ROCmSoftwarePlatform/rccl.git && \
cd rccl && git checkout 6707a27 && \
mkdir build && cd build && \
CXX=/opt/rocm/bin/hipcc cmake -DCMAKE_INSTALL_PREFIX=/opt/rocm/rccl .. && \
CXX=/opt/rocm/bin/hipcc cmake -DCMAKE_INSTALL_PREFIX=/usr/local .. && \
make -j ${NUM_MAKE_RCCL} && make install && \
cd /tmp && \
rm -rf rccl
Expand Down

0 comments on commit 496f4e2

Please sign in to comment.