博客
关于我
libtorch c++ 调用(一)源码编译与测试
阅读量:541 次
发布时间:2019-03-09

本文共 1110 字,大约阅读时间需要 3 分钟。

编译并使用PyTorch的项目需要注意Visual Studio版本与CUDA版本的匹配,同时确保依赖库正确加载。以下是从你提供的信息整理出来的优化后的步骤,希望对你有所帮助:

  • 检查Visual Studio版本:确保安装的是Visual Studio 2017 15.6或更高版本,并确认《MSC VER》≤1912。可以通过右键点击项目文件,选择“属性”,查看工具set版本。

  • 升级CUDA工具set:如果需要支持CUDA 10.x版本,确保安装了CUDA 10.2及其以上版本的工具set。可以通过NVIDIA的网站下载相应的安装包。

  • 下载适配的libtorch:在PyTorch官网下载对应CUDA版本和Visual Studio版本的预编译libtorch库。例如,若使用CUDA 10.2和Visual Studio 2017 15.6,可以下载对应的libtorch1.5.1-cu10.2.

  • 配置编译环境

    • 打开命令提示符,切换到你要编译的项目目录。
    • ensure CMake版本≥3.12.
    • 设置生成器为Visual Studio 2017,并指定对应的工具set版本(e.g., 15.6)。
    • 设置CMAKE_GENERATOR_TOOLSET_VERSION为14.11,以匹配你的Visual Studio工具set。
  • 编译libtorch

    • 使用cmake生成项目文件。
    • 运行cmake --build .,生成编译结果。
    • 检查生成文件的路径,确保使用了正确的编译选项(如Release或Debug模式)。
  • 生成测试项目代码

    • 从网络获取并修改为你的需求。确保导入必要的库文件和头文件(如torch/script.hat.h)。
    • 调整代码以适应PyTorch脚本模块的执行方式,如使用torch::jit::load加载.pt文件。
  • 处理依赖库冲突

    • 检查生成的可执行文件的依赖,确保所有必要的库和DLL(如c10.dlltorch_cpu.dll)已经被正确链接。
    • 使用dumpbin工具详细查看每个依赖文件,确保没有缺失。
  • 调试和优化

    • 如果运行时出错,检查是否有错误提示,并根据文档解决。
    • 使用调试工具指_ERROR Reporting检查代码问题。
    • 确保生成的可执行文件路径正确,环境变量路径设置正确。
  • 验证测试/demo的运行性

    • 在命令行运行可执行文件,查看输出结果。
    • 输入测试数据,确保PyTorch模块能够正确运行。
  • 通过逐步检查每个环节,包括依赖设置、编译选项、动态库路径等,应该可以解决主流问题。遇到问题时,参考社区讨论或PyTorch官方文档,确保配置与环境完全兼容。

    转载地址:http://dtqiz.baihongyu.com/

    你可能感兴趣的文章
    NIFI1.21.0通过Postgresql11的CDC逻辑复制槽实现_指定表或全表增量同步_实现指定整库同步_或指定数据表同步配置_04---大数据之Nifi工作笔记0056
    查看>>
    NIFI1.23.2_最新版_性能优化通用_技巧积累_使用NIFI表达式过滤表_随时更新---大数据之Nifi工作笔记0063
    查看>>
    NIFI从MySql中增量同步数据_通过Mysql的binlog功能_实时同步mysql数据_根据binlog实现数据实时delete同步_实际操作04---大数据之Nifi工作笔记0043
    查看>>
    NIFI从MySql中增量同步数据_通过Mysql的binlog功能_实时同步mysql数据_配置binlog_使用处理器抓取binlog数据_实际操作01---大数据之Nifi工作笔记0040
    查看>>
    NIFI从MySql中增量同步数据_通过Mysql的binlog功能_实时同步mysql数据_配置数据路由_实现数据插入数据到目标数据库_实际操作03---大数据之Nifi工作笔记0042
    查看>>
    NIFI从MySql中增量同步数据_通过Mysql的binlog功能_实时同步mysql数据_配置数据路由_生成插入Sql语句_实际操作02---大数据之Nifi工作笔记0041
    查看>>
    NIFI从MySql中离线读取数据再导入到MySql中_03_来吧用NIFI实现_数据分页获取功能---大数据之Nifi工作笔记0038
    查看>>
    NIFI从MySql中离线读取数据再导入到MySql中_不带分页处理_01_QueryDatabaseTable获取数据_原0036---大数据之Nifi工作笔记0064
    查看>>
    NIFI从MySql中离线读取数据再导入到MySql中_无分页功能_02_转换数据_分割数据_提取JSON数据_替换拼接SQL_添加分页---大数据之Nifi工作笔记0037
    查看>>
    NIFI从PostGresql中离线读取数据再导入到MySql中_带有数据分页获取功能_不带分页不能用_NIFI资料太少了---大数据之Nifi工作笔记0039
    查看>>
    nifi使用过程-常见问题-以及入门总结---大数据之Nifi工作笔记0012
    查看>>
    NIFI分页获取Mysql数据_导入到Hbase中_并可通过phoenix客户端查询_含金量很高的一篇_搞了好久_实际操作05---大数据之Nifi工作笔记0045
    查看>>
    NIFI分页获取Postgresql数据到Hbase中_实际操作---大数据之Nifi工作笔记0049
    查看>>
    NIFI同步MySql数据_到SqlServer_错误_驱动程序无法通过使用安全套接字层(SSL)加密与SQL Server_Navicat连接SqlServer---大数据之Nifi工作笔记0047
    查看>>
    NIFI同步MySql数据源数据_到原始库hbase_同时对数据进行实时分析处理_同步到清洗库_实际操作06---大数据之Nifi工作笔记0046
    查看>>
    Nifi同步过程中报错create_time字段找不到_实际目标表和源表中没有这个字段---大数据之Nifi工作笔记0066
    查看>>
    NIFI大数据进阶_FlowFile拓扑_对FlowFile内容和属性的修改删除添加_介绍和描述_以及实际操作---大数据之Nifi工作笔记0023
    查看>>
    NIFI大数据进阶_FlowFile生成器_GenerateFlowFile处理器_ReplaceText处理器_处理器介绍_处理过程说明---大数据之Nifi工作笔记0019
    查看>>
    NIFI大数据进阶_FlowFile生成器_GenerateFlowFile处理器_ReplaceText处理器_实际操作---大数据之Nifi工作笔记0020
    查看>>
    NIFI大数据进阶_Json内容转换为Hive支持的文本格式_实际操作_02---大数据之Nifi工作笔记0032
    查看>>