Windows从源码构建tensorflow(离线编译)

由一开始的在线编译,到后面的离线编译,一路踩坑无数,历经整整6个半小时,终于编译成功!在此记录一下参考过的文章,有时间整理一下踩坑记录。

一、环境配置

tensorflow官网上有版本对应关系

win10

bazel 3.1.0

msys2

tensorflow2.3.0

python3.5-3.8

MSVC2019

protobuf3.9.2 //protobuf的版本貌似非常重要,在tensorflow/tensorflow/workspace.bzl文件中可以查找到对应的版本

cuda10.1

cudnn7.6.5

VS2019 //编译过程中好像因为VS版本问题报过错,查找VS的MSVC版本,在报错文件中会有要求

二、编译前的准备工作,包括下载安装软件,配置环境变量,设置虚拟环境等

三、开始编译

bash 复制代码
bazel build --config=opt --config=cuda //tensorflow:tensorflow_cc.dll --local_ram_resources=200 --local_cpu_resources=10

四、编译报错处理

C++ compilation of rule '//tensorflow/core/kernels:XXX' failed

在windows环境下离线编译tensorflow2.2.0的C++动态链接库

Windows10 Bazel 编译 Tensorflow 2 C++ dll 和 lib 文件

Windows编译Tensorflow2.2.0 C++ dll (CPU bazel)

五、在VS上测试

上面的步骤完成后, 在 G:\tensorflow_c++_2.3(在你放源码的地方) 下的 bazel_bin 中会有相应的 include 文件夹 , tensorflow_cc.dlltensorflow_cc.lib 文件, 把它们复制出来放到你的程序里就可以了.


Window下搭建Tensorflow的C++环境

相关推荐
齐翊31 分钟前
怎么确认 AI 看懂了你的提示词?
人工智能·github·ai编程
饼干哥哥1 小时前
Reddit VOC调研太慢?搭一个AI专家团队半小时洞察任何品类|以猫用饮水机为例
人工智能·算法·ai编程
以和为贵1 小时前
前端也能搞懂 RAG:用 JS 手写一条最小检索增强链路
前端·人工智能·面试
武子康2 小时前
调查研究-192 AI Agent 之间也需要“信任“:把多 Agent 信任变成可测指标
人工智能·openai·agent
Smoothcloud_润云2 小时前
Hermes Agent 的上下文记忆机制:一个开源 Agent 是怎么"记住"你的
人工智能·agent·gpu
早点睡啊2 小时前
精读 LangChain 官方文档(一)总览、安装与快速开始:从 create_agent 跑通第一个智能体
人工智能
牛奶3 小时前
AI时代裁员后:清零是君子豹变
人工智能·程序员
武子康3 小时前
调查研究-191 SenseVoice 不只是 ASR:把语音从“转文字“升级成“理解状态“
人工智能·深度学习·openai
程序员cxuan4 小时前
Codex 会把磁盘给烧了?完整复盘来了!
人工智能·后端·程序员