libtorch是 pytorch c++ 前端, 是pytorch的一个c++接口
就是可以用c++来写pytorch
libtorch(pytorch c++)的大多数api和pytorch保持一致
libtorch是pytorch的C++版本,支持CPU端和GPU端的部署和训练。由于python和c++的语言特性,因此用pytorch做模型训练,libtorch做模型部署。
用libtorch部署pytorch模型,而不是用tensorrt等工具部署模型的优势在于:pytorch和libtorch同属一个生态,API语句比较接近,并且不会出现某网络层不支持的问题。
但是libtorch不依赖于python。python训练的模型需要转换为script model才能由libtorch加载,并进行推理。
libtorch (pytorch c++前端 / pytorch的C++版本)
于 2021-11-21 19:30:57 首次发布