Skip to content

GitLab

Unverified Commit d19142f6 authored by danielhua23's avatar danielhua23 Committed by GitHub
Browse files

[AMD] Fix 3 bugs when build docker on amd mi3x gpu (#1401)

parent f2858fa1
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 3 deletions
docker/Dockerfile.rocm
View file @ d19142f6
...@@ -22,10 +22,12 @@ RUN conda run -n py_3.10 conda install pip cmake -y && \ ...@@ -22,10 +22,12 @@ RUN conda run -n py_3.10 conda install pip cmake -y && \
RUN apt-get install -y python3 python3-dev python3-setuptools gcc libtinfo-dev zlib1g-dev build-essential cmake libedit-dev libxml2-dev RUN apt-get install -y python3 python3-dev python3-setuptools gcc libtinfo-dev zlib1g-dev build-essential cmake libedit-dev libxml2-dev
RUN git clone https://github.com/tile-ai/tilelang.git --recursive -b main tilelang && \ RUN git clone https://github.com/tile-ai/tilelang.git --recursive -b main tilelang && \
conda run -n py_3.10 bash -c "cd tilelang && USE_ROCM=1 pip install -e . -v" mv /opt/conda/envs/py_3.10/compiler_compat /opt/conda/envs/py_3.10/compiler_compat.bak || true && \
conda run -n py_3.10 bash -c "pip install 'numpy<2.0' --force-reinstall && cd tilelang && USE_ROCM=1 pip install -e . -v"
RUN conda init bash RUN conda init bash && \
echo "conda activate py_3.10" >> /root/.bashrc
SHELL ["/bin/bash", "-l", "-c"] SHELL ["/bin/bash", "-l", "-c"]
CMD ["bash", "-c", "source ~/.bashrc && conda activate py_3.10 && exec bash"] ENTRYPOINT ["/bin/bash", "--login", "-i"]
\ No newline at end of file \ No newline at end of file
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