在RK3588 GPU 安装TVM
发布网友
发布时间：2024-10-10 06:38
共1个回答
热心网友
时间：2024-11-13 10:56
RK3588搭载的ARM Mali G610 GPU，具备约500GFLOPS的单精度运算能力，并支持OpenCL 2.1。本次将通过编译安装带有OpenCL功能的TVM，并在RK3588上运行模型。
配置OpenCL编译选项是关键步骤。首先，在RK3588系统中使用`sudo find / -name *mali.so`查找libmali.so，这是包含OpenCL相关操作的动态库。然后，在build/config.cmake文件中，将USE_OPENCL项目填写为libmali.so的地址。
接着，将OpenCL头文件目录CL放置于`/usr/include`中，并将路径添加到`~/.bashrc`的PATH变量中。执行命令于tvm的build目录下进行编译。
在完成编译后，需先卸载已有的TVM版本。具体操作为：通过`pip3 uninstall tvm`命令进行卸载，接着进入源码目录`apache-tvm-src-v0.10.0/python`，删除指定的目录与文件。之后，进入tvm目录继续清理，确保环境整洁。
安装TVM的过程如下：进入`apache-tvm-src-v0.10.0/python`目录执行相关安装命令。最后，通过验证步骤确认安装正确，查看输出结果，以确保TVM在RK3588 GPU上运行正常。
参考官方文档以获取更多详细信息：tvm.apache.org/docs/install/
