cmake_minimum_required(VERSION 3.10)  # CMAKE的最低版本要求
set(CMAKE_CXX_STANDARD 17)  # 使用的C++标准
set(CMAKE_CXX_STANDARD_REQUIRED ON)

project(hello VERSION 1.0)  # 项目名称 版本 版本号
set(TORCH_PATH /home/wty/Programs/mambaforge/envs/jax/lib/python3.11/site-packages/torch/share/cmake)
find_package(Torch REQUIRED PATHS ${TORCH_PATH})

add_executable(torch_tensor torch_tensor.cpp)  # 可执行文件名称（目标） 联合编译的文件 [文件1, 文件2, ...]
target_link_libraries(torch_tensor ${TORCH_LIBRARIES})
