栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > C/C++/C#

配置C++/Tensorflow环境

C/C++/C# 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

配置C++/Tensorflow环境

系统环境:
Ubuntu 20.04
Cuda 11.3
Cudnn 8
Python3.8
gcc9.3

Step 1.
google下载最新版Tensorflow 2.6.2
根据"C++调用Tensorflow”,需安装protobuf,absl,eigen3,bazel
需要注意版本问题,因此需参考tensorflow下的workspace.bzl文件中所提及的版本来安装对应版本的包,版本过高会导致编译问题

Step 2.
Protobuf安装

#cd protobuf
#git submodule update --init --recursive
#./autogen.sh
#./configure
#make
#make check
#make install
#ldconfig

运行protoc --version,输出版本信息则表示安装成功

Step 3.
下载对应版本eigen3以及abseil-cpp,cp -r所有*.h及*.so到对应的/usr/local/include或/usr/local/lib下

Step 4.
安装对应版本的bazel

Step 5.
在tensorflow子目录下

./configure
bazel build --config=opt --config=cuda //tensorflow/tools/pip_package:build_pip_package //tensorflow:libtensorflow_cc.so

编译后将所需的*.so及*.h分别cp -r至/usr/local/lib及/usr/local/include中

至此完成全部安装

测试:

#include 
#include 

#include 

using namespace std;
using namespace tensorflow;

int main()
{
    Session* session;
    Status status = NewSession(SessionOptions(), &session);
    if (!status.ok()) {
        cout << status.ToString() << "n";
        return 1;
    }
    cout << "Session successfully created.n";
}

CMakeLists.txt部分

cmake_minimum_required (VERSION 2.8.8)
project (main)
 
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -g -std=c++11 -W")

//link_directories(/usr/local/lib/tensorflow)

add_executable(main main.cpp) 
target_link_libraries(main tensorflow_cc tensorflow_framework)
转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/702257.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号