Visual Studio导入libtorch(Cuda版)

Visual Studio导入libtorch(Cuda版)

一、安装

官网:https://pytorch.org/get-started/locally/

相应地选择并下载

二、环境变量配置

解压zip,得到libtorch文件夹,将libtorch\liblibtorch\bin对应路径添加到系统环境变量-Path中。

这里加入的是最后两个libtorch相关的,顺便找到之前的cuda路径,之后在vs配置时有用

三、VS配置

(1)右键项目名->属性【或者Alt+Enter快捷键调出面板】

  • 设置为x64平台

(2)VC++目录

  1. 包含目录

    选择箭头->编辑

    复制代码
    # 填入自己对应的cuda和libtorch相关路径
    CUDA路径\CUDA\v11.7\include
    libtorch路径\libtorch\include
    libtorch路径\libtorch\include\torch\csrc\api\include
  2. 库目录编辑

    复制代码
    CUDA路径\CUDA\v11.7\lib\x64
    libtorch路径\libtorch\lib

(3)C\C++

  1. 常规-附加包含目录

    和包含目录一样

    复制代码
    CUDA路径\CUDA\v11.7\include
    libtorch路径\libtorch\include
    libtorch路径\libtorch\include\torch\csrc\api\include

    也可以直接用**$(IncludePath)**替代这三行

  2. 语言

    • 如果报错#error You need C++17 to compile PyTorch,那么C++语言标准选择ISO C++17 标准 (/std:c++17)

    • 如果报错#error std 不明确的符号,那么把符合模式改为否

  3. 预编译头

    • 设置为不使用预编译头,否则可能会出现#include xxx 无法编译的问题

(4)链接器

  1. 常规-附加库目录

    和库目录一样

    复制代码
    CUDA路径\CUDA\v11.7\lib\x64
    libtorch路径\libtorch\lib
  2. 输入-附加依赖项

    • 路径\libtorch\lib中检索cuda和torch,将对应的所有lib加入到附加依赖项中,以及c10等;

    • 路径\libtorch\lib下要加入cudart.lib

    主要是以下:

    复制代码
    # 路径\libtorch\lib
    c10.lib
    c10_cuda.lib
    pytorch_jni.lib
    torch.lib
    torch_cuda.lib
    torch_cpu.lib
    # torch_cuda_cpp.lib
    # torch_cuda_cu.lib
    
    # 路径\CUDA\v11.7\lib
    cudart.lib
  3. 命令行

    复制代码
    指令一: # 适用cuda11.8版本、libtorch1.8.1
    /INCLUDE:?warp_size@cuda@at@@YAHXZ
    
    指令二: # 适用libtorch1.9.0
    /INCLUDE:"?searchsorted_cuda@native@at@@YA?AVTensor@2@AEBV32@0_N1@Z 
    
    指令三: # 适用1.10.2+cu113版本
    /INCLUDE:?warp_size@cuda@at@@YAHXZ -INCLUDE:?searchsorted_cuda@native@at@@YA?AVTensor@2@AEBV32@0_N1@Z 

四、报错

1、由于找不到xx.dll,无法继续执行代码。重新安装程序可能会解决此问题

解决方案:在调试-环境中加入

复制代码
# PATH=libtorch路径\libtorch\lib;%PATH%
PATH=F:\Software\Alib\libtorch\lib;%PATH%

参考链接:

C++下的libtorch部署------(三)window10下的Visual Studio配置libtorch

C++部署Pytorch(Libtorch)出现问题、错误汇总

相关推荐
滕州市燕猫虎计算机科技工作室个体工商户9 小时前
IDEA:Command line is too long
java·ide·intellij-idea
Hhy_110719 小时前
《C++深度解构04》类和对象(下)——类型转换、static、友元与编译器优化
c语言·c++·学习·类和对象·visual studio
free-elcmacom1 天前
C语言字符指针与字符串字面量——从一道“剑指 Offer”笔试题说起
c语言·visual studio
辰辉创聚2 天前
重组蛋白聚集的形成机制及其对蛋白活性的影响
opencv·时序数据库·visual studio·重组蛋白·蛋白聚集·蛋白活性·蛋白聚集体
AI编码进化论2 天前
云IDE介绍:环境模板化、Agent上云,云IDE该怎么选
开发语言·ide·人工智能·团队开发·ai编程
我要见SA姐12 天前
DeepSeek Harness 开源贡献手记:从 Issue 到 Merge 的完整旅程
ide·vscode·python·自动化·编辑器
VIP_CQCRE3 天前
在 Visual Studio 里接入 Ace Data Cloud:用 LMLocal 快速获得 AI 编程能力
openai·ai编程·visual studio·ace data cloud·lmlocal
ThornArmor3 天前
《向内深潜,向外飞掠》
c语言·开发语言·c++·vim·visual studio
郝学胜_神的一滴3 天前
C++20模板元编程 04:变量模板 别名模板与模板Lambda实战
c++·visual studio
2501_915918413 天前
使用 VS Code 写 Swift 做 iOS 开发,插件能实现哪些功能,还有哪些短板
ide·vscode·ios·objective-c·个人开发·swift·敏捷流程