TX2入门教程软件篇-安装TensorFlow1.0.1(jetpack3.0)

TX2入门教程软件篇-安装TensorFlow

说明：

• 介绍如何在TX2上安装深度学习框架TensorFlow

准备：

• 利用Jetpack安装如下：

  • L4T 27.1 an Ubuntu 16.04 64-bit variant (aarch64)
  • CUDA 8.0
  • cuDNN 5.1.10

• TensorFlow安装需要用到CUDA和cuDNN

• TensorFlow占用比较多空间，TX2通常空间不足，最好增加64G+的U盘作为root分区启动，增加交换分区大小为8G+

• 或者挂载U盘作为分区，移植大文件或目录到U盘，并做好ln软连接到原来的目录位置。

• 总体测试下来，安装时间比较长，会有错误出现，根据错误做修复。

• 如果入口脚本命令报错的话，请根据脚本里面的命令一条条执行，并处理错误信息。并可以注释下载类命令，再重复执行。

• 安装的版本为：TensorFlow v1.0.1

安装：

• 下载脚本：

mkdir ~/dl
cd ~/dl
git clone https://github.com/ncnynl/installTensorFlowTX2.git
cd installTensorFlowTX2

• 安装依赖：

chmod +x installPrerequisites.sh
 ./installPrerequisites.sh

• 包含安装java，Protobuf，grpc-java ，Bazel等
• 下载tensorflow代码

$ ./cloneTensorFlow.sh

• 设置环境变量

$ ./setTensorFlowEV.sh

• 编译TensorFlow

$ ./buildTensorFlow.sh

• 编译时间比较长，中间会有自动退出的情况，继续执行命令直到完成
• 打包成whl文件，放在$HOME目录下,如：tensorflow-1.0.1-cp27-cp27mu-linux_aarch64.whl

$ ./packageTensorFlow.sh

• 安装whl文件

$ pip install $HOME/tensorflow-1.0.1-cp27-cp27mu-linux_aarch64.whl

• 提示权限问题的话，pip前加sudo

测试：

• 运行TensorFlow例子：

$ cd ~/dl/installTensorflowTX2/
$ vim test.py

• 内容如下：

#!/usr/bin/env python 

import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))

• 效果如下：

$ chmod +x test.py
$ python test.py

...................
Hello, TensorFlow!

问题集：

• 问题1：提示找不到http://zlib.net/zlib-1.2.8.tar.gz
• 解决：需要打补丁，cloneTensorFlow.sh里面的打补丁没执行好

• 问题2：提示junit.4.12.jar不能下载
• 解决：修改vim ~/tensorflow/tensorflow/workspace.bzl 更换地址，并去掉校验

  native.http_jar(
      name = "junit_jar",
      #url = "https://github.com/junit-team/junit4/releases/download/r4.12/junit-4.12.jar",
      url = "https://github.com/orrsella/bazel-example/blob/master/third_party/junit/junit-4.12.jar",
      #sha256 = "59721f0805e223d84b90677887d9ff567dc534d7c502ca903c0c2b17f05c116a",
      #sha256 = "fe3d4c56388dc3d74049abae83f4520f6703062e174e16bb5551cdf439ca4f81",
  )

参考：

• https://www.tensorflow.org/get_started/os_setup.html
• https://devtalk.nvidia.com/default/topic/901148/tensorflow-on-jetson-tx1/?offset=11
• https://github.com/tensorflow/tensorflow/
• https://github.com/tensorflow/tensorflow/issues/851
• http://www.jetsonhacks.com/2016/12/30/tensorflow-nvidia-jetson-tx1-development-kit/
• https://github.com/jetsonhacks/installTensorFlowTX1
• http://www.jetsonhacks.com/2016/12/30/install-tensorflow-on-nvidia-jetson-tx1-development-kit/
• http://www.jetsonhacks.com/2017/01/15/tensorflow-build-update-jetson-tx1/