如何在Visual Studio 2019中启动并配置一个使用pyTorch的C++项目（Windows系统，CMAKE项目）

首先感谢做这个视频的小哥：

https://www.youtube.com/watch?v=6eTVqYGIWx0&t=165s&ab_channel=PythonEngineer

写下文章备忘。

首先打开VS后，新建一个CMAKE项目，起名为“demoPytorch”。（小哥说因为CMAKE配置pyTorch比较简单）

在这个页面下载pyTorch：https://pytorch.org/get-started/locally/ 

选择Windows、LibTorch、C++、显卡对应的CUDA版本.。然后选择debug版本下载。

下载后解压。

然后打开项目文件夹中的CMakeLists.txt文件（注意不是out文件夹），并加入下面这一行：

find_package(Torch REQUIRED)

告诉CMake我们需要Torch这个包。此时Ctrl+S保存，出现错误，无法找到这个包。

点开下拉栏：

点击管理配置。

在CMake命令参数（CMake command argument）中写入：

-DCMAKE_PREFIX_PATH="C:\Users\Think\Documents\Visual Studio 2019\libtorch"

（填入解压后的pyTorch路径，注意换成双反斜\）

告诉CMake在哪里可以找到pyTorch。

输入后，Ctrl+S保存。如果你已经安装了CUDA，这时候应该显示CMake正常配置。如果没有安装CUDA，会报错，请下载安装CUDA后把CMake变量中的CUDA_SDK_ROOT_DIR设置为CUDA dev安装后的路径。

之后，在cpp文件中加入：

#include "torch/torch.h"

现在include下应该有红线，我们需要到CMakeLists.txt中进行库链接。

加入：

target_link_libraries(demoPytorch "${TORCH_LIBRARIES}")

Ctrl+S后，回到cpp文件，红线消失。

验证一下是否配置成功，在cpp中把cout那一行改成：

    cout << torch::randn({3,2}) << endl;

运行出错，提示找不到c10.dll文件。

参考：https://pytorch.org/cppdocs/installing.html

复制以下代码到CMakeLists.txt文件：

# The following code block is suggested to be used on Windows.
# According to https://github.com/pytorch/pytorch/issues/25457,
# the DLLs need to be copied to avoid memory errors.
if (MSVC)
  file(GLOB TORCH_DLLS "${TORCH_INSTALL_PREFIX}/lib/*.dll")
  add_custom_command(TARGET example-app
                     POST_BUILD
                     COMMAND ${CMAKE_COMMAND} -E copy_if_different
                     ${TORCH_DLLS}
                     $<TARGET_FILE_DIR:example-app>)
endif (MSVC)

注意把example-app改为自己的项目名。

Ctrl+S后，等待CMake配置成功的提示，然后再次运行，成功：

 附上CMakeLists.txt文件：

# CMakeList.txt: demoPytorch 的 CMake 项目，在此处包括源代码并定义
# 项目特定的逻辑。
#
cmake_minimum_required (VERSION 3.8)

find_package(Torch REQUIRED)

# 将源代码添加到此项目的可执行文件。
add_executable (demoPytorch "demoPytorch.cpp" "demoPytorch.h")
target_link_libraries(demoPytorch "${TORCH_LIBRARIES}")

if (MSVC)
  file(GLOB TORCH_DLLS "${TORCH_INSTALL_PREFIX}/lib/*.dll")
  add_custom_command(TARGET demoPytorch
                     POST_BUILD
                     COMMAND ${CMAKE_COMMAND} -E copy_if_different
                     ${TORCH_DLLS}
                     $<TARGET_FILE_DIR:demoPytorch>)
endif (MSVC)

# TODO: 如有需要，请添加测试并安装目标。

.cpp文件:

// demoPytorch.cpp: 定义应用程序的入口点。
//

#include "demoPytorch.h"
#include "torch/torch.h"

using namespace std;

int main()
{
    cout << torch::randn({3,2}) << endl;
    return 0;
}