libtorch (pytorch c++前端 / pytorch的C++版本)

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加载,并进行推理。

你可能感兴趣的:(pytorch,C++)