Pytorch部署方案(二):利用LibTorch(C++库)来调用训好的模型【trace跟踪PyTorch训练好的模型-->用Libtorc调用PyTorch训练好的模型】

利用C++来调用PyTorch训练好的模型,其实也是利用官方强大的LibTorch库。

一、LibTorch的安装

虽然说安装,其实就是下载官方的LibTorch包而已,从官方网站中选择PyTorch(1.1),libtorch,以及cuda的版本,其中会出现下载链接,这里为cuda9.0的链接:https://download.pytorch.org/libtorch/cu90/libtorch-shared-with-deps-latest.zip。下载好找个路径解压。解压完放在那不动!!

二、PyTorch模型训练

这里我使用了最为简单ResNet50的预训练模型,其中保存跟踪模型的代码如下:

import torch
import torchvision.models as models
from PIL import Image
import numpy as np
image = Image.

你可能感兴趣的:(AI/工程部署,pytorch,深度学习,c++,LibTorch)