Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[ci] Add a Dockerfile for building manylinux2014-compatible Taichi wheels with CUDA backend #4491

Merged
merged 2 commits into from
Mar 9, 2022
Merged

[ci] Add a Dockerfile for building manylinux2014-compatible Taichi wheels with CUDA backend #4491

Changes from 1 commit
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
05ab214
[ci] Add a Dockerfile for building manylinux2014-compatible Taichi wh…
strongoier Mar 9, 2022
4f1097b
Apply suggestions
strongoier Mar 9, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
49 changes: 49 additions & 0 deletions ci/Dockerfile.manylinux2014.cuda
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,49 @@
FROM nvidia/cudagl:11.2.2-devel-centos7

LABEL maintainer="https://github.com/taichi-dev"

# Install development tools with expected versions
strongoier marked this conversation as resolved.
Show resolved Hide resolved
RUN yum install -y git wget epel-release centos-release-scl
RUN yum install -y cmake3 devtoolset-10-gcc*
RUN ln -s /usr/bin/cmake3 /usr/bin/cmake
ENV PATH="/opt/rh/devtoolset-10/root/usr/bin:$PATH"

# Build LLVM/Clang 10 from source
WORKDIR /
RUN wget https://github.com/llvm/llvm-project/releases/download/llvmorg-10.0.0/llvm-10.0.0.src.tar.xz
RUN tar -xf llvm-10.0.0.src.tar.xz && rm llvm-10.0.0.src.tar.xz
RUN wget https://github.com/llvm/llvm-project/releases/download/llvmorg-10.0.0/clang-10.0.0.src.tar.xz
RUN tar -xf clang-10.0.0.src.tar.xz && rm clang-10.0.0.src.tar.xz
RUN cp -r clang-10.0.0.src llvm-10.0.0.src/tools/clang

WORKDIR /llvm-10.0.0.src/build
RUN cmake .. -DLLVM_ENABLE_RTTI:BOOL=ON -DBUILD_SHARED_LIBS:BOOL=OFF -DCMAKE_BUILD_TYPE=Release -DLLVM_TARGETS_TO_BUILD="X86;NVPTX" -DLLVM_ENABLE_ASSERTIONS=ON -DLLVM_ENABLE_TERMINFO=OFF
RUN make -j 8 && make install
ENV CC="/usr/local/bin/clang"
ENV CXX="/usr/local/bin/clang++"

# Link gcc 10 to build Taichi
WORKDIR /usr/lib/gcc/x86_64-redhat-linux/
RUN ln -s /opt/rh/devtoolset-10/root/usr/lib/gcc/x86_64-redhat-linux/10 10
# Check gcc-10 is used
RUN clang++ -v

# Create non-root user for running the container
RUN useradd -ms /bin/bash dev
WORKDIR /home/dev
USER dev

# Install miniconda
RUN wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh && \
bash Miniconda3-latest-Linux-x86_64.sh -p /home/dev/miniconda -b
ENV PATH="/home/dev/miniconda/bin:$PATH"

# Set up multi-python environment
RUN conda init bash
RUN conda create -n py36 python=3.6 -y
RUN conda create -n py37 python=3.7 -y
RUN conda create -n py38 python=3.8 -y
RUN conda create -n py39 python=3.9 -y
strongoier marked this conversation as resolved.
Show resolved Hide resolved

WORKDIR /home/dev
ENV LANG="C.UTF-8"