hienanh/tvm repository overview

FROM ubuntu:20.04
ENV DEBIAN_FRONTEND noninteractive

WORKDIR /root
RUN apt update
RUN apt install -y wget git python3 python3-pip python3-dev python3-setuptools gcc libtinfo-dev zlib1g-dev build-essential cmake libedit-dev libxml2-dev gcc libtinfo-dev zlib1g-dev build-essential cmake libedit-dev libxml2-dev llvm-6.0 clang-6.0 lldb-6.0 lld-6.0 libgomp1 zip unzip && rm -rf /var/lib/apt/lists/*
RUN python3 -m pip install numpy pandas tornado xgboost==1.1.0 torch torchvision cloudpickle psutil spell kaggle tokenizers transformers pytest

RUN git clone --recursive https://github.com/apache/tvm tvm
WORKDIR /root/tvm
RUN git submodule init
RUN git submodule update --init --recursive --jobs 0
RUN git reset --hard 7b3a22e465dd6aca4729504a19beb4bc23312755
RUN git submodule update --recursive
RUN mkdir -p build && cp cmake/config.cmake build
RUN cp cmake/config.cmake build
RUN mv build/config.cmake build/~config.cmake
RUN cat build/~config.cmake | sed -E "s|set\(USE_GRAPH_RUNTIME OFF\)|set\(USE_GRAPH_RUNTIME ON\)|" | sed -E "s|set\(USE_GRAPH_EXECUTOR OFF\)|set\(USE_GRAPH_EXECUTOR ON\)|" | sed -E "s|set\(USE_PROFILER OFF\)|set\(USE_PROFILER ON\)|" | sed -E "s|set\(USE_RELAY_DEBUG OFF\)|set\(USE_RELAY_DEBUG ON\)|" | sed -E "s|set\(USE_LLVM OFF\)|set\(USE_LLVM /usr/bin/llvm-config-6.0\)|" > build/config.cmake

WORKDIR /root/tvm/build
RUN cmake .. && make -j$(nproc)
WORKDIR /root/tvm/python
RUN python3 setup.py install --user
WORKDIR /root/tvm/nnvm
RUN make -j$(nproc)
WORKDIR /root/tvm/3rdparty/dmlc-core
RUN make -j$(nproc)
WORKDIR /root/tvm/3rdparty/dlpack
RUN make -j$(nproc)

WORKDIR /root