Ubuntu下C++调用pytorch训练好模型--利用libtorch
最近因项目需要,需C++调用pytorch模型,以下是一些学习心得,把过程记录下来。
搭建有以下几个流程:
官网下载库libtorch
pytorch模型转化
编写C++调用程序
一、官网下载库libtorch
下载libtorch地址(https://pytorch.org/get-started/locally/),选择对应版本(根据自身电脑配置),比如我是Ubuntu+CUDA 10.2
下载好之后,选择路径进行解压(注意自己下载的包),命令如下:
unzi
原创
2020-07-22 11:12:30 ·
3516 阅读 ·
2 评论