Unverified Commit f21bfef2 authored by pnunna93's avatar pnunna93 Committed by GitHub
Browse files

Dockerfile: Remove fixed rccl version in rocm5.1.x docker file (#476)

**Description**
The commit(e08b6d3a) installs a rccl
version which is causing "undefined symbol: ncclGetLastError" while
trying to import torch. Revert it to avoid the error.
parent 121a5ddc
......@@ -114,16 +114,6 @@ RUN cd /tmp && \
cp ./Linux/mlc /usr/local/bin/ && \
rm -rf ./Linux mlc.tgz
# Install rccl with commitid 6707a27
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=/usr/local .. && \
make && make install && \
cd /tmp && \
rm -rf rccl
ENV PATH="${PATH}:/opt/rocm/hip/bin/" \
LD_LIBRARY_PATH="/usr/local/lib/:${LD_LIBRARY_PATH}" \
SB_HOME=/opt/superbench \
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment