ubuntu20.04 tvm 安装教程

ubuntu20.04 tvm 安装教程:

参考:
1. https://tvm.hyper.ai/docs/install/from_source/
2. https://blog.csdn.net/wenwen_2020/article/details/134856293

步骤:

1. 创建容器:docker run -itd --name tvm --gpus=all --ipc=host -p 20322:22 -v /mnt/iro/tt:/home/tt ubuntu:20.04 /bin/bash -c "while true; do echo \'Hello\'; sleep 1000; done"

2. apt update
   apt-get install -y python3 python3-dev python3-setuptools gcc libtinfo-dev zlib1g-dev build-essential libedit-dev libxml2-dev
   
3. 安装cmake:
   wget https://github.com/Kitware/CMake/releases/download/v3.28.3/cmake-3.28.3.tar.gz
   tar -zxvf cmake-3.28.3.tar.gz
   apt-get install libssl-dev
   ./bootstrap
   make
   make install
   ln -s /home/tt/tools/cmake-3.28.3/bin/* /usr/bin/
   cmake -version
   
4. apt install libncurses5
   wget https://github.com/llvm/llvm-project/releases/download/llvmorg-13.0.0/clang+llvm-13.0.0-x86_64-linux-gnu-ubuntu-20.04.tar.xz
   tar -xvf clang+llvm-13.0.0-x86_64-linux-gnu-ubuntu-20.04.tar.xz
   mv clang+llvm-13.0.0-x86_64-linux-gnu-ubuntu-20.04 llvm13
   vim /etc/profile
        最后添加两行:
            export LLVM_HOME=/home/tt/tools/llvm13/bin
            export PATH=$LLVM_HOME:$PATH
    source /etc/profile
    llvm-as --version
    
5. 
set(USE_LLVM OFF)
改为:
set(USE_LLVM ON) 或者 SET(USE_LLVM /home/tt/tools/llvm13/bin/llvm-config)

set(USE_CUDA OFF)
改为:
set(USE_CUDA ON) 或者 SET(USE_CUDA /cuda的路径)

set(USE_CUDNN OFF) 改为:set(USE_CUDNN ON)

set(USE_GRAPH_EXECUTOR OFF) 改为:set(USE_GRAPH_EXECUTOR ON)

set(USE_PROFILER OFF) 改为:set(USE_PROFILER ON)

6. 
cd build
cmake ..
make -j24

7. 
vim ~/.bashrc
# 文件末尾添加这几行
export TVM_HOME=/home/tt/project/tvm
export PYTHONPATH=$TVM_HOME/python:$TVM_HOME/python/tvm:$TVM_HOME/build:$TVM_HOME/python/tvm/runtime:$[PYTHONPATH]


注意:
1. 官方导出TVM_LOG_DEBUG有bug,import tvm时会报错,修改为:
export TVM_LOG_DEBUG="ir/transform.cc=1,relay/ir/transform.cc=1"

2. ubuntu20.04的话,注意用13.0.0的llvm,不然会core dump!!!!!

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值