Ubuntu18.04 配置 LibTorch

2023-12-29 07:23| 来源: 网络整理| 查看: 265

cmake 编译构建应用程序，目录的布局如下：在这里插入图片描述在 example/ 文件夹中新建 build 文件夹，用来编译程序：

$ mkdir build $ cd build

执行编译命令：

#cmake -DCMAKE_PREFIX_PATH=/absolute/path/to/libtorch .. $ cmake -DCMAKE_PREFIX_PATH=/XXXX/libtorch ..

在这里插入图片描述因为我使用了cuda，这里会有一个 warning ,但是已经编译成功了，不用管它。其中/absolute/path/to/libtorch是解压缩的LibTorch的绝对路径，如果通过conda或pip安装了PyTorch，则可以使用torch.utils.cmake_prefix_path变量查询 CMAKE_PREFIX_PATH 。在这种情况下，CMake配置步骤如下所示：

$ cmake -DCMAKE_PREFIX_PATH=`python -c 'import torch;print(torch.utils.cmake_prefix_path)'`

然后执行命令生成可执行文件：

$ cmake --build . --config Release

在这里插入图片描述最后执行命令测试：

$ ./example-app

【本文地址】

公司简介

联系我们

今日新闻

推荐新闻

专题文章