如何通过CMake将PyTorch头文件安装到include的项目子目录下

方法一：构建配置时直接指定 #

在执行CMake配置命令时，通过组合CMAKE_INSTALL_PREFIX和CMAKE_INSTALL_INCLUDEDIR变量，直接指定头文件的目标子目录结构：

cmake -S <pytorch源码目录> -B <构建目录> \
  -DCMAKE_INSTALL_PREFIX=<你的安装根目录> \
  -DCMAKE_INSTALL_INCLUDEDIR=include/pytorch

执行后，安装的头文件会自动放到<安装根目录>/include/pytorch/torch下，完全匹配你的期望结构。如果想换成libtorch作为子目录名，把CMAKE_INSTALL_INCLUDEDIR的值改成include/libtorch即可。

方法二：修改PyTorch的CMake安装规则 #

如果需要更灵活的定制，可以直接修改PyTorch源码中的CMake安装脚本：

1. 定位到PyTorch源码中负责头文件安装的INSTALL命令（通常在torch/CMakeLists.txt或相关子模块配置文件里）。
2. 原命令大致如下：

   INSTALL(DIRECTORY torch/
           DESTINATION ${CMAKE_INSTALL_INCLUDEDIR}
           FILES_MATCHING PATTERN "*.h" PATTERN "*.hpp")
   

3. 修改DESTINATION参数，添加自定义子目录前缀：

   INSTALL(DIRECTORY torch/
           DESTINATION ${CMAKE_INSTALL_INCLUDEDIR}/pytorch
           FILES_MATCHING PATTERN "*.h" PATTERN "*.hpp")
   

4. 重新执行CMake配置、构建和安装流程即可生效。

方法三：安装后手动调整（临时方案） #

如果只是临时需求，不想修改构建配置，可以在安装完成后手动调整目录结构：

# 示例：假设原安装路径为/usr/local/include，目标路径为/usr/local/include/pytorch
mkdir -p /usr/local/include/pytorch
mv /usr/local/include/torch /usr/local/include/pytorch/

注意：这种方法只是临时解决，后续重新安装PyTorch时会恢复原结构，长期使用建议用前两种CMake配置的方式。

内容的提问来源于stack exchange，提问作者codesniffer