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 ·
3438 阅读 ·
2 评论