TVM 安装笔记

TVM install

Posted by Treaseven on December 12, 2024

安装依赖

  • CMake (>=3.24.0)
  • LLVM (recommended >= 15)
  • conda
  • gcc 12.4.0 g++ 12.4.0
  • python (>=3.8)

官网教程

安装官网教程

按照官网教程安装,不过在编译的时候会出现问题
内存不够:将内存设置在20G,共享内存设置在8G
gcc编译版本混乱: 使用anaconda环境,意味着同时存在系统的gcc/g++和anaconda环境中的gcc/g++;LLVM库文件可能是用系统的编译器编译的,而你在用anaconda环境中的编译器编译TVM,导致链接时找不到正确版本的标准库函数
解决方法:设置conda环境的编译器,在conda环境安装gcc-12

conda remove gcc gxx libgcc libgcc-ng libstdcxx libstdcxx-ng  # 删除当前版本
conda install -c conda-forge gcc=12 gxx=12 libgcc-ng=12 libstdcxx-ng=12  # 安装gcc-12
gcc --version  # 验证安装
g++ --version
export CC=$(which gcc)  # 设置编译器路径
export CXX=$(which g++)

虚拟机不能安装cuda和cudnn: set(USE_CUDA OFF)
声明TVM安装路径: 安装教程在这两行需要修改

export TVM_PATH=/home/xxx/tvm   # xxx代表你自己tvm文件夹所在的目录
export PYTHONPATH=$TVM_PATH/python:$TVM_PATH/python/tvm:$TVM_PATH/build:$TVM_PATH/python/tvm/runtime:${PYTHONPATH}