TX2上搭建tensorflow

TX2上搭建tensorflow

TX2作为Nvidia强大的边缘计算工具，为了更好地进行AI开发，比如在上面跑一个神经网络进行图像识别等，Nvidia官方为TX2提供了专门的tensorflow，甚至开发了对应的tensorrt，用来加速推理，因此，只要刷好机，在TX2上安装tensorflow并不难。注：TX2详细的刷机教程见我的另一篇博客。

一、TX2上装tensorflow和win10上装的区别

先说一下TX2和台式计算机的区别，它们的最大区别就是计算机架构不一样，TX2是ARM架构，而通常的台式计算机是AMD架构，这是安装tensorflow需要不同方法的根本原因。还有一个区别是操作系统的不同，TX2刷机需要jetson SDK，里面的原生OS（操作系统）是ubuntu系统，而在我的另一篇文章中（最简）win10上搭建tensorflow环境（gpu或cpu多版本）用的是win10操作系统。

二、TX2安装tensorflow

安装东西前，都要进行源的刷新：

sudo apt-get update

参考Nvidia官方文档，TX2作为Nvidia的一个产品，官方有提供比较详尽的开发文档，因此参考资料要以官方为准。

根据刷机时装的不同版本jetback SDK进行不同版本的安装。tensorflow版本的查看可以进入https://developer.download.nvidia.com/compute/redist/jp/v44（这个网址来自上面的文档）查看。
注意一：安装的时候最好只使用pip3 install指令，而不用sudo pip3 install指令，减少对其他安装环境的影响。
注意二：jetpack SDK的版本与对应的tensorflow版本不一定要完全和上图一致，如jetpack4.5.1也可以安装tensorflow2.2。
注意三：安装过程中可能会遇到一些小问题，如scipy-1.4.1不能正常安装的问题，这个问题是因为目前官网已经不支持scipy-1.4.1的安装，解决问题的办法参考这个链接;可能也会遇到非法指令核心已转储的问题，这个问题是因为python2和python3的numpy版本冲突引起的，解决问题的办法参考这个链接

三、安装成功的检验

最后一步，同样进行安装成功的检验，终端输入python3，刷完机的TX2中通常由python2和python3两个环境。然后输入import tensorflow as tf，回车运行，如果没有报错则证明安装成功，恭喜。
查看版本，可继续输入：

tf.__version__

查看gpu版本是否安装成功，输入：

tf.test.is_gpu_available()

如果输出True，恭喜安装gpu版本成功了！