NVIDIA Jetson TX2(Xavier/NANO) 刷Jetpack4.4.1 Python3.6 安装TensorFlow1.15.X/2.X PyTorch Keras...

很多教程都不全或是总是报错…以下分享给需要的朋友。

=================================================

0. 准备工作

1. 一台X86架构的HOST主机（就一般笔记本或PC），安装Ubuntu16.04（高版本比如20.04不行，18.04应该可以，我用的是16.04）；
2. NVIDIA Jetson TX2；
3. USB-MicroUSB 用来连接HOST主机和TX2；
4. 确保主机和TX2可以在同一个网络下。

1. 在X86主机上安装NVIDIA SDK Manager

1. 下载SDK Manager：在NVIDIA官网下载
2. 下载完成之后是一个.deb文件，安装：

sudo apt install ./sdkmanager-[version].[build#].deb 

安装完成后可以在terminal中输入sdkmanager启动。

2. 刷Jetpack4.4.1

需要登录，使用的是英伟达开发的账户，需要注册一个。

登录后加入STEP1，这里用USB-MicroUSB连接主机和TX2，点击refresh就可以找到TX2，不用自己选。

然后一直继续按提示完成刷机即可。

中途有可能会出现下载失败、安装失败什么什么的，没关系，重新打开重新来就行，支持断点续传，搞几次就好了。

FLASH完成后，需要输入TX2的IP地址、用户名、密码，安装要求完成就可以。这里说一下，有的教程说用手动模式，其实直接用自动就可以，要注意修改一下TX2的IP地址，修改为正确的地址就可以。

完成刷机后，就到安装TensorFlow了，后面就不用主机了，在TX2上操作。

这里要先说一下：

不要改apt-get的下载源！
不要改apt-get的下载源！
不要改apt-get的下载源！

重要的事情说三遍。

3. 安装TensorFlow

在TX2中，Ctrl + Alt + T 打开terminal，输入python3 --version 查看python版本。

我的是3.6.x，那就要去安装对应的TensorFlow版本。

官方有提供Jetpack4.4.1 + python3.6.x 对应的TensorFlow版本，可以在这里查看：TensorFlow for Jetson TX2!

其他版本的硬件、Python不同版本，对应的TensorFlow也可以在这里找得到：

不同硬件

不同软件

这边是Python 3.6+JetPack4.4，所以按照Python 3.6+JetPack4.4的相关安装流程安装就可以：

为什么前面不让改源，因为我发现改了之后这块总是报错，我也不清楚为什么，反正我没改就弄好了，只是浪费了些时间。

sudo apt-get install libhdf5-serial-dev hdf5-tools libhdf5-dev zlib1g-dev zip libjpeg8-dev liblapack-dev libblas-dev gfortran
sudo apt-get install python3-pip
sudo pip3 install -U pip
sudo pip3 install -U pip testresources setuptools numpy==1.16.1 future==0.17.1 mock==3.0.5 h5py==2.9.0 keras_preprocessing==1.0.5 keras_applications==1.0.8 gast==0.2.2 futures protobuf pybind11
# TF-2.x
$ sudo pip3 install --pre --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v44 tensorflow==2.3.1+nv20.11
# TF-1.15
$ sudo pip3 install --pre --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v44 ‘tensorflow<2’

首先，这里安装的时候，我用普通用户安装会报错，文件夹权限问题，建议直接用root权限安装

这里有个问题，就是会遇到这样的错误：

BUILDING WHEEL FOR H5PY (SETUP.PY) ... ERROR或 ERROR: FAILED BUILDING WHEEL FOR H5PY

解决办法：

先安装Cython即可：

sudo pip3 install Cython

sudo pip3 install h5py

这里应该是这样安装的，如果不对…我忘了，大家想想办法哈哈哈哈哈哈…

还有就是，安装TensorFlow1.15的时候，‘tensorflow<2’这里会报错，建议直接去这里确认要安装的版本，然后像安装2.X一样把下载补充完整，例如：

sudo pip3 install --pre --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v44 tensorflow==1.15.4+nv20.11

其他的问题就百度一下吧。应该问题不大。

安装完成，测试一下，在terminal中输入python3打开交互模式，输入import tensorflow as tf 没有报错就可以了。

上述相关下载也可以直接下载文件包，然后离线安装。

4. 安装PyTorch/Keras/Caffe2/…

有兴趣的同学可以参考Jetson Zoo，包括安装各种ML框架、ROS、CV等等，很全很有用，建议收藏！

还有这是NVIDIA官方论坛里的PyTorch相关安装说明也可以参考一下！

这里介绍一下安装PyTorch：

JetPack 4.4 / 4.4.1

PyTorch v1.6.0 pip wheel (Python 3.6)
PyTorch v1.7.0 pip wheel (Python 3.6)

# install OpenBLAS and OpenMPI
$ sudo apt-get install libopenblas-base libopenmpi-dev

# Python 3.6 (download pip wheel from above)
$ sudo apt-get install python3-pip
pip3 install Cython
pip3 install numpy torch-1.6.0-cp36-cp36m-linux_aarch64.whl #这里根据你实际下的whl来安装

用pip3来安装的话，有可能会遇到没有科学上网的问题：

WARNING: Retrying (Retry(total=4, connect=None, read=None, redirect=None, status=None)) after connection broken by 'NewConnectionError('<pip._vendor.urllib3.connection.HTTPSConnection object at 0x7fa405b278>: Failed to establish a new connection: [Errno 101] Network is unreachable',)': /simple/typing-extensions/

可以手动修改pip安装源：

阿里云 http://mirrors.aliyun.com/pypi/simple/

豆瓣(douban) http://pypi.douban.com/simple/

清华大学 https://pypi.tuna.tsinghua.edu.cn/simple/

中国科学技术大学 http://pypi.mirrors.ustc.edu.cn/simple/

这样改：

pip3 install numpy torch-1.6.0-cp36-cp36m-linux_aarch64.whl -i https://pypi.tuna.tsinghua.edu.cn/simple/

就可以啦，如果还有错，那就百度一下吧…

可以参考一下这篇文章：pip安装时遇到的问题

这里说了一句：note — the PyTorch and Caffe2 projects have merged, so installing PyTorch will also install Caffe2

所以Caffe2框架也安好了~

=============================================

有问题可以留言。

加油，打工人！

其他可以参考：ROS学习01-NVIDIA JETSON TX2 使用Jetpack4.4刷机指南，这个兄弟建议改源，大家也可以尝试一下。