如何在CmakeLists.txt中添加 tf 包
时间: 2024-02-11 21:03:46 浏览: 211
要在 CMakeLists.txt 中添加 TensorFlow 包,可以按照以下步骤进行操作:
1. 在项目的根目录下,创建一个名为 `cmake` 的文件夹,并在其中创建一个名为 `FindTensorFlow.cmake` 的文件。
2. 在 `FindTensorFlow.cmake` 文件中添加以下内容:
```
find_package(TensorFlow REQUIRED)
if(TensorFlow_FOUND)
message(STATUS "Found TensorFlow ${TensorFlow_VERSION}")
else()
message(FATAL_ERROR "TensorFlow not found")
endif()
```
3. 在项目的 `CMakeLists.txt` 文件中添加以下内容:
```
cmake_minimum_required(VERSION 3.10)
project(your_project_name)
set(CMAKE_CXX_STANDARD 11)
# 添加 TensorFlow 包
set(CMAKE_MODULE_PATH ${PROJECT_SOURCE_DIR}/cmake)
find_package(TensorFlow REQUIRED)
# 添加你的源码文件
add_executable(your_executable_name your_source_files.cpp)
# 链接 TensorFlow 库
target_link_libraries(your_executable_name TensorFlow::TensorFlow)
```
这样,你就可以在你的 C++ 项目中使用 TensorFlow 包了。
阅读全文