如何在 windows 配置 libtorch c++ 前端库?下载 pytorch 已经编译好的库:此库不带 gpu,主要方便演示。支持 win7 win10 系统。下载地址:https://download.pytorch.org/libtorch/cpu/libtorch-win-shared-with-...
如何在 windows 配置 libtorch c++ 前端库?
下载 pytorch 已经编译好的库:
此库不带 gpu,主要方便演示。支持 win7 win10 系统。
下载地址:https://download.pytorch.org/libtorch/cpu/libtorch-win-shared-with-deps-latest.zip
1. cmake 配置
1.1 新建 CMakeLists.txt
并添加以下内容:
# 设置 cmake 版本限制
cmake_minimum_required(VERSION 3.0 FATAL_ERROR)
# 项目名称
project(libtorch-app)
# 设置 libtorch-win-shared-with-deps-latest 目录,主要让 find_package 可以找到 Torch 的 TorchConfig.cmake 配置文件以及其他相关 Config.cmake 配置文件
set(CMAKE_PREFIX_PATH "./libtorch-win-shared-with-deps-latest")
find_package(Torch REQUIRED)
add_executable(libtorch-app main.cpp)
target_link_libraries(libtorch-app "${TORCH_LIBRARIES}")
set_property(TARGET libtorch-app PROPERTY CXX_STANDARD 11)
1.2 在 CMakeLists.txt
同级目录新建一个 main.cpp
文件,添加以下内容:
#include <torch/torch.h>
#include <iostream>
int main()
{
torch::Tensor tensor = torch::rand({ 9,9 });
std::cout << tensor << std::endl;
return 0;
}
1.3 然后在 CMakeLists.txt
同级目录下打开一个命令行(按住 Shift + 鼠标右键即可
)输入以下命令:
cmake -DCMAKE_BUILD_TYPE=Release -G "Visual Studio 14 Win64"
执行完以上命令后生成 libtorch-app.sln
解决方案文件,打开编译即可。
2. 手动配置(仅适用于CPU,GPU需要自行另外添加相关依赖)
新建一个属性页并添加当当前配置文件中
2.1. 在 C/C++->常规->附加包含目录里面添加以下内容并开启多处理器编译:
$(SolutionDir)3rdparty\libtorch-win-shared-with-deps-latest\include\torch\csrc\api\include
$(SolutionDir)3rdparty\libtorch-win-shared-with-deps-latest\include
2.2. 在 连接器->输入->附加依赖项里面添加以下内容:
$(SolutionDir)3rdparty\libtorch-win-shared-with-deps-latest\lib\torch.lib
$(SolutionDir)3rdparty\libtorch-win-shared-with-deps-latest\lib\caffe2.lib
$(SolutionDir)3rdparty\libtorch-win-shared-with-deps-latest\lib\c10.lib
2.3. 在 调试->环境 里面添加环境变量
path=%path%;$(SolutionDir)3rdparty\libtorch-win-shared-with-deps-latest\lib
2.4 新建一个 main.cpp 文件,添加以下内容并使用 x64 模式编译即可:
#include <torch/torch.h>
#include <iostream>
int main()
{
torch::Tensor tensor = torch::rand({ 9,9 });
std::cout << tensor << std::endl;
return 0;
}
输出一个随机数值9x9
矩阵:
`0.2407 0.9294 0.5167 0.3774 0.9841 0.6530 0.9825 0.5814 0.4903 0.8882 0.5111 0.7414 0.7563 0.1666 0.2542 0.2624 0.0668 0.4328 0.0481 0.4880 0.6299 0.5140 0.0379 0.9187 0.3033 0.1510 0.6705 0.1983 0.6113 0.2893 0.0700 0.8585 0.3588 0.3891 0.0551 0.0458 0.7738 0.0797 0.0611 0.1781 0.3898 0.0238 0.0361 0.3905 0.2005 0.5774 0.5769 0.6275 0.3511 0.9609 0.3415 0.7188 0.5650 0.5670 0.0828 0.2139 0.5793 0.4089 0.5725 0.3938 0.8250 0.2695 0.6470 0.4106 0.3609 0.9982 0.8789 0.0134 0.8454 0.3880 0.0937 0.9598 0.1523 0.9423 0.2989 0.6404 0.0997 0.2817 0.0706 0.1867 0.1980 Variable[CPUFloatType]{9,9} ]
本文标题为:如何在 windows 配置 libtorch c++ 前端库?
- C语言qsort()函数的使用方法详解 2023-04-26
- Easyx实现扫雷游戏 2023-02-06
- c++ const 成员函数,返回一个 const 指针.但是返回的指针是什么类型的 const? 2022-10-11
- 详解C语言中sizeof如何在自定义函数中正常工作 2023-04-09
- C语言手把手带你掌握带头双向循环链表 2023-04-03
- Qt计时器使用方法详解 2023-05-30
- C++ 数据结构超详细讲解顺序表 2023-03-25
- C语言详解float类型在内存中的存储方式 2023-03-27
- ubuntu下C/C++获取剩余内存 2023-09-18
- 我应该为我的项目使用相对包含路径,还是将包含目录放在包含路径上? 2022-10-30