Visual Studio导入libtorch（Cuda版）

Visual Studio导入libtorch（Cuda版）

一、安装

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

相应地选择并下载

二、环境变量配置

解压zip，得到libtorch文件夹，将libtorch\lib和libtorch\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
   1
   2
   3
   4

2. 库目录编辑

   CUDA路径\CUDA\v11.7\lib\x64
   libtorch路径\libtorch\lib
   1
   2

（3）C\C++

1. 常规-附加包含目录

   和包含目录一样

   CUDA路径\CUDA\v11.7\include
   libtorch路径\libtorch\include
   libtorch路径\libtorch\include\torch\csrc\api\include
   1
   2
   3

   也可以直接用**$(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
   1
   2

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
   1
   2
   3
   4
   5
   6
   7
   8
   9
   10
   11
   12

   

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
   2
   3
   4
   5
   6
   7
   8

   

四、报错

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

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

# PATH=libtorch路径\libtorch\lib;%PATH%
PATH=F:\Software\Alib\libtorch\lib;%PATH%
1
2

参考链接：

C++下的libtorch部署——（三）window10下的Visual Studio配置libtorch

C++部署Pytorch（Libtorch）出现问题、错误汇总