TensorFlow 源代码调试与修改 VScode

最新推荐文章于 2024-08-09 11:36:53 发布

Frank123721

最新推荐文章于 2024-08-09 11:36:53 发布

阅读量2.3k

点赞数 1

分类专栏：深度学习文章标签： TensorFlow 源代码调试修改 VScode

本文链接：https://blog.csdn.net/zhaoyunxiang721/article/details/105556346

版权

本文详细介绍了如何使用VScode调试和修改TensorFlow源代码，包括安装依赖、配置、编译、创建launch.json文件，以及如何进行代码修改和添加自定义操作。作者分享了个人经验，特别强调VScode在调试过程中的优势，并提供了在线课程链接，帮助读者深入理解TensorFlow内核。

摘要由CSDN通过智能技术生成

TensorFlow是目前使用最广泛的深度学习框架，网上关于如何进行TensorFlow编程的教程或者博客比比皆是。然而，对于那些希望了解TensorFlow内部如何实现，或者是希望自定义修改TensorFlow内核的朋友来说，搜索相关的内容却很难找到有帮助的教程或者博客，不论是中文还是外文。笔者去年花费好几个月对如何调试TensorFlow源代码，再到如何修改TensorFlow内核C++库文件，最后自定义TensorFlow内部的Tensor形状以及卷积模式，较熟悉如何进行TensorFlow的源代码调试与修改。本文将介绍如何对TensorFlow进行调试，进而便于有需要的朋友进行内核（C++）的学习与修改。
网上有少许关于如何调试TensorFlow源代码的文章，有的推荐直接用gdb，有的推荐用Emacs+gdb，有的推荐用codeblock的，笔者通过实验发现VS code是可视化做的最好的，而且用户界面友好，调试不会出现卡顿。Emacs在调试时随着追踪的深入会卡顿；而单纯用gdb…可能适合于linux大佬；codeblock用户界面不及VS code，而且笔者尝试调试没有成功…。

一：准备工作：

首先安装VS code，python，gcc，bazel等必要库。
然后git下载TensorFlow的源代码并check out出自己需要的版本。对源代码进行configure，根据是否有GPU而进行相应选择。这里需要注意：TensorFlow的版本必须要和本机的python，gcc，bazel版本相匹配，建议提前想好再安装所有库文件。
在configure源文件之后，我们