Relay外部库使用 本文介绍如何将cuDNN或cuBLAS等外部库与Relay一起使用. Relay内部使用TVM生成目标特定的代码.例如,使用cuda后端,TVM为用户提供的网络中的所有层生成cuda内核.有时将各种供应商开发的外部库合并到Relay中也很有帮助.幸运的是,TVM具有透明地调用这些库的机制.对于Relay用户,要做的只是适当地设置目标字符串. 在可以使用Relay的外部库之前,TVM必须与要使用的库一起构建.例如,要使用cuDNN,需要启用cmake / config.cma