源码:
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