本文提供了TVMDL的详细安装步骤,包括从源码下载、解压、配置环境到最终编译的过程。特别强调了如何在3rd文件夹中设置依赖,以及在config.cmake中启用LLVM和CUDA的方法。
源码：
wget http://tvm.d2l.ai/d2l-tvm.zip
unzip d2l-tvm.zip -d d2l-tvm
也可以到GitHub搜索tvm，然后需要把在3rd文件夹里面空的依赖下载好之后放在该文件夹
然后：
mkdir build
cp config.cmake build/
gedit build/config.cmake
设置：
set(USE_LLVM ON)
# 如果想使用cuda
# set(USE_CUDA ON)
最后：
cmake ..
make -j16
您可能感兴趣的与本文相关的镜像
PyTorch 2.5
PyTorch 是一个开源的 Python 机器学习库，基于 Torch 库，底层由 C++ 实现，应用于人工智能领域，如计算机视觉和自然语言处理
2186
661
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
扫一扫
