每日归档： 2023年5月21日

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

Posted on by remaper in 深入浅出 tvm
本文面向 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，都在镜像内完成