Jetson TX2 入门 ——安装tensorflow

时隔四天，tensorflow是终于安装好了，刷机完之后天真的以为tensorflow很好安装，结果被TX2狠狠的打了脸，总之还是自己学艺不精。

刷完机，老师要求在TX2上安装python3.7版本，装好之后也把python3指向的默认版本改成了python3.7，接下来就是安装tensorflow，搞了四天，真的崩溃了，第一次安装tensorflow以为随便找个版本就好，结果就下载了一个找到的最高的版本tensorflow1.13，结果运行pip安装之后，错误一大堆，查了很多资料后才发现jetpack3.2刷机的TX2只能安装1.5-1.8，并且TX2所需要的tensorflow是aarch，和一般linux安装的大不相同，好的，这些东西理清楚后，又发现网上的版本只有python2.7和python3.5的tensotflow，而我的默认python已经设置为python3.7，无奈又得重新改回python3.5，但是改着改着，就发现许多依赖包没了，pip的python2版本又下载不了，于是，，，又重新刷机了，这次改为jetpack3.3刷机。

一、重新刷机（jetpack3.3）

    刷机步骤参考以下博客：

    https://blog.csdn.net/nulidehahafr/article/details/97754593

二、安装tensorflow

 看了网上的安装方式，大概有两种：直接安装和编译安装。我第一次用的编译安装，但是需要内存很大，要分一个8G的虚拟内存，我运行到最后一步出了一个错误，但一直没找到解决方法，就换成直接安装。因此就只说一下直接安装。

 1、安装cuda9.0

    昨天看博客有博主说刷机后在TX2上并没有安装好cuda和cudnn，所以一般得再次安装一遍。我们刷机时，主机上有个jetpack_download文件夹，里面就有cuda和cudnn的安装文件，因此，我们把这个文件夹直接拷贝到TX2上，然后运行以下命令：

//打开新终端
sudo apt-key add /var/cuda-repo-9-0-local/7fa2af90.pub
//cd进入jetpack_download文件夹
sudo dpkg -i cuda-repo-l4t-9-0-local_9.0.252-1_arm64.deb
sudo dpkg -i libcudnn7_7.1.5.14-1+cuda9.0_arm64.deb
sudo dpkg -i libcudnn7-dev_7.1.5.14-1+cuda9.0_arm64.deb
sudo apt update
sudo apt install cuda-toolkit-9.0

 运行nvcc -V，若输出cuda的相关信息，则安装成功，如果提示没有nvcc指令，那就从头再来吧。。。

2、安装pip3

   TX2刷机之后自带python2.7和python3.5，我安装的tensorflow是基于python3.5的，因此需要安装pip3.

   https://bootstrap.pypa.io/在该链接中下载get-pip.py，运行

   sudo python3 get-pip.py

   再用pip -V测试pip是否安装好

3、安装tensorflow

进入NVIDIA官网（如下链接）下载tensorflow_gpu-1.9.0

 https://developer.nvidia.com/embedded/downloads#?search=tensorflow&tx=$product,jetson_tx2

//安装系统包
sudo apt-get install libhdf5-serial-dev hdf5-tools libhdf5-dev zlib1g-dev zip libjpeg8-dev

//安装python包

sudo pip3 install -U numpy grpcio absl-py py-cpuinfo psutil portpicker six mock requests gast h5py astor termcolor protobuf keras-applications keras-preprocessing wrapt google-pasta

sudo apt-get install python-dev python3-dev

//安装tensorflow
sudo pip3 install tensorflow_gpu-1.9.0+nv18.8-cp35-cp35m-linux_aarch64.whl

 

  在安装tnesorflow的时候，在ubuntu里下载的是一个压缩包，就直接用压缩包安装，千万别解压之后再安装，因为解压之后文件后缀就会变成.whl_FILE，安装这个会出现很多错误。（今天查了一下tensorflow官网，Google使用wheel标准分发tensorflow，它就是.whl后缀的ZIP格式文件，并且tensorflow推荐使用pip install）我就是这样安装了好几天，错误在网上也找不到解决方法。并且在这一步网络状态一定要好，昨天早上装了一早上都是网络错误，吃了饭过来再试了一次就一下安装好了。

4、验证tensorflow

打开一个终端，运行python3，然后导入tensorflow，不报错就表示安装成功

import tensorflow as tf

tf.__version__

输出：1.9.0

5、卸载tensorflow（这步是你需要卸载它的时候再运行）

pip3 uninstall -y tensorflow-gpu

6、升级tensorflow

pip3 install--upgrade tensorflow