Dockerfile: Update rccl version and fix issue in rocm5.1.1 dockerfile (#354)
**Description** Update rccl version and fix issue in rocm5.1.1 dockerfile.
This commit is contained in:
Родитель
81a4146bc1
Коммит
e08b6d3a1c
|
@ -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
|
||||
|
||||
|
|
Загрузка…
Ссылка в новой задаче