ROCm环境下安装tvm

什么也没留下

已于 2022-12-12 19:36:48 修改

阅读量1.2k

点赞数

文章标签： python

于 2022-06-15 11:56:34 首次发布

本文链接：https://blog.youkuaiyun.com/weixin_43971060/article/details/125293745

版权

本文档详细介绍了在ROCm环境下安装TVM库的过程，包括配置cmake、python、gcc和ROCm LLVM环境，修改config.cmake文件指定相关路径，解决cmake构建时找不到ROCM的问题，以及编译、安装和验证TVM的步骤。在测试阶段，针对C++测试超时问题提出了解决方案，并提供了测试效果的展示。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

ROCm环境下安装tvm主要是为了构建libtvm.so、libtvm_runtime.so共享库

配置cmake环境，建议3.14版本以上
配置python环境，建议python3版本以上
配置gcc环境、ROCm的llvm环境
从下载页面https://tvm.apache.org/docs/install/index.html从下载 tvm 源代码，解压tar xvzf apache-tvm-src-v0.8.0.tar.gz（此处可将源码包修改名字成tvm）

cd tvm 
mkdir build && mkdir install
cd build  
cp ../cmake/config.cmake ./

修改config.cmake文件：
a) 指定ROCm路径：set(USE_ROCM /opt/rocm-4.x)
b) 指定LLVM路径（此处一定要到llvm-config二进制码）：set(USE_LLVM /opt/rocm-4.x/llvm/bin/llvm-config)
c) set(USE_MIOPEN ON)、set(USE_ROCBLAS ON)、set(USE_GRAPH_EXECUTOR ON)、set(USE_PROFILER ON)、set(USE_RELAY_DEBUG ON)
cmake -DCMAKE_BUILD_TYPE=Debug -DCMAKE_INSTALL_PREFIX=/home/tvm/install ..
我这里在cmake构建时报错：“Cannot find ROCM, USE_ROCM=/opt/rocm-4.x ”（将USE_ROCM=ON改为自动搜索也不成）→ 在源码中搜索关键字发现是tvm/cmake/modules/R

最低0.47元/天解锁文章