日期: 2023 年 5 月 21 日

深入浅出 tvm – (2) 搭建开发环境

Posted on by remaper in 深入浅出 tvm Tagged:
本文面向 tvm 开发者，粗略介绍一下 tvm 的安装过程

1. 开发环境

系统：ubuntu 20.04，由于中文生态兼容的比较好，比如输入法，有道云笔记，所以这个系统本身也是我的主力开发系统
tvm 版本：v0.10.0，如果你的不是这个版本，记得 checkout 一下，否则下面的依赖可能不全，对于学习 tvm 来说，v0.10.0 版本其实够了

2. 依赖安装

以我的 dockerfile 为例，可以参考
另外，大家也可以跳过这一步，直接用我弄好的镜像：docker pull pipul/tvm_ci_cpu
FROM ubuntu:20.04

ENV TZ=Europe/Kiev
RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo $TZ > /etc/timezone

COPY ./apt /etc/apt
COPY ./etc/sudoers.d/90-nopasswd-sudo /etc/sudoers.d/90-nopasswd-sudo
COPY ./.bashrc /etc/bashrc

RUN apt-get update
RUN apt-get install -y vim sudo git g++ pip llvm
RUN apt-get install -y python3 python3-dev python3-setuptools gcc libtinfo-dev zlib1g-dev build-essential cmake libedit-dev libxml2-dev
RUN pip install onnx numpy attrs decorator Pillow psutil scipy relpath typing-extensions tornado 'xgboost>=1.1.0' cloudpickle
有几个不是必须的：
  1. COPY ./apt 主要是替换了163的源，不然软件安装太慢
  2. /etc/sudoers.d/90-nopasswd-sudo 是 tvm 编译的时候必须依赖的，但是 docker 镜像 ubuntu:20.04 是默认没有的，所以需要搞下
  3. COPY ./.bashrc 可以忽略
假设构建好的镜像为：${tvm-docker-image}
一旦镜像构建好了之后，后续编译和开发 tvm，都在镜像内完成