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++环境

相关推荐
天上掉下来个程小白6 分钟前
Stream流的中间方法
java·开发语言·windows
暮雪倾风12 分钟前
【WPF开发】控件介绍-Grid(网格布局)
windows·wpf
DogDaoDao27 分钟前
【预备理论知识——2】深度学习:线性代数概述
人工智能·深度学习·线性代数
牛哥带你学代码28 分钟前
交叠型双重差分法
人工智能·深度学习·机器学习
学步_技术35 分钟前
自动驾驶系列—线控系统:驱动自动驾驶的核心技术解读与应用指南
人工智能·机器学习·自动驾驶·线控系统·转向系统
jmoych43 分钟前
AI正悄然地影响着企业数字化转型
人工智能
说私域1 小时前
社群团购中的用户黏性价值:以开源小程序多商户AI智能名片商城源码为例
人工智能·小程序
深度学习实战训练营2 小时前
基于keras的停车场车位识别
人工智能·深度学习·keras
乔代码嘚2 小时前
AI2.0时代,普通小白如何通过AI月入30万
人工智能·stable diffusion·aigc
墨@#≯2 小时前
机器学习系列篇章0 --- 人工智能&机器学习相关概念梳理
人工智能·经验分享·机器学习