Advertisement

libtorch版本2基于VC加载模型进行推理过程

阅读量:

此前已成功构建适用于libtorch的模型,接下来将在VC2015环境下完成相应的部署工作。

基础运行环境配置如下:
操作系统版本为Windows 10;
所使用的编译工具为Visual Studio 2015;
CUDA相关组件版本为CUDA 10配合cuDNN 7.6.5;
Python解释器版本为3.6.1;
PyTorch框架版本为1.4.0;
Libtorch库版本同样为1.4.0。

1.libtorch下载与配置

若使用的是最新版本,建议直接从官方提供的链接进行下载:
https://pytorch.org/get-started/locally/

本人所使用的为1.4.0版本,其获取路径如下:
https://download.pytorch.org/libtorch/cu101/libtorch-win-shared-with-deps-1.4.0.zip

完成下载之后,对文件进行解压操作,即可获得相应内容:

在这里插入图片描述

2.新建VC工程

在进行文件包含操作时,只需将 include 文件夹直接复制即

全部评论 (0)

还没有任何评论哟~