配置libtorch

本文详细介绍了如何配置libtorch,并在C++中使用PyTorch模型。首先,需要安装nvidia显卡驱动、CUDA和CuDNN。接着,下载并解压libtorch,将模型通过torch.jit.trace转换为torch.jit.ScriptModule。在C++中,利用torch/script.h加载模型,并编写CMakeLists.txt文件进行编译。完成编译后,运行生成的可执行文件即可调用模型。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

  1. 根据系统,安装nvidia显卡驱动、cuda、cudnn
    下载链接:
    cuda下载:https://developer.nvidia.com/cuda-downloads
    cudnn下载:https://developer.nvidia.com/rdp/cudnn-archive
    cudnn安装指导:https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html
  2. 下载安装libtorch (gpu)
    参考: https://pytorch.org/cppdocs/installing.html
    2.1 根据系统及需求选择cuda版本的libtorch
    下载链接:https://pytorch.org/

命令行运行
wget https://download.pytorch.org/libtorch/cu102/libtorch-shared-with-deps-1.8.1%2Bcu102.zip
2.2 解压
命令行运行
unzip libtorch-shared-with-deps-1.8.1+cu102.zip
将解压得到的文件放到 "~/libtorch"目录

  1. 编写、运行c++代码
    参考: https://pytorch.apachecn.org/docs/1.0/cpp_export.html
    3.1 将pytorch模型转化为c++可调用的模型
    运行python代码:
    import torch
    import torchvision

get model

model = torchvision.models.resnet18()<

### 如何使用CMake配置LibTorch #### 创建项目结构 为了成功配置并编译带有 LibTorch 的 C++ 项目,首先需要创建合适的项目文件夹结构。假设项目的根目录已经包含了 `CMakeLists.txt` 和源代码文件(如 `example.cpp`),接下来应该在该目录下建立一个新的子文件夹用于存放构建产物。 ```bash mkdir build cd build ``` #### 配置CMake 一旦进入了新创建的 `build` 文件夹内,就可以运行 CMake 来生成所需的构建系统文件。对于 Windows 用户来说,如果打算利用 Visual Studio 进行开发,则可以在 Developer Command Prompt for VS 2022 中执行如下命令来指定 LibTorch 安装位置: ```batchfile cmake -DCMAKE_PREFIX_PATH=D:\LibTorch\libtorch\share\cmake\Torch .. ``` 这里 `-DCMAKE_PREFIX_PATH` 参数指定了 LibTorch 库的位置[^2]。注意路径应替换为实际安装路径。 #### 构建解决方案 当上述命令顺利完成后,将会自动生成适用于所选 IDE 或者工具链的工程文件。如果是针对 MSBuild 使用的话,继续在同一终端中输入下面这条指令以完成最终的编译过程,并将输出放置于 Release 子目录之下: ```batchfile msbuild OXI_Model_Project.sln /p:Configuration=Release /m ``` 这段脚本会调用 Microsoft Build Tools 对整个方案进行处理,从而得到可以独立运行的应用程序二进制文件。 #### 设置IDE集成(可选) 对于那些更倾向于图形界面操作而非纯命令行工作的开发者而言,在 Visual Studio 2019 中可以通过编辑 `CMakeSettings.json` 文件来进行更多定制化调整。开启显示高级变量功能可以帮助确认所有的 CMake 变量已被正确定义;而任何在顶层 `CMakeLists.txt` 中声明过的宏定义都将优先级高于 JSON 文件内的设定[^3]。 另外,如果你正在使用 Qt Creator 开发环境,记得前往其“构建环境”设置页面添加 LibTorch 头文件所在路径,以便让 IDE 正确识别外部依赖库中的函数原型和其他接口信息[^4]。 #### 关联问题探讨
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值