【玩转Jetson TX2 NX】（四）TX2 NX安装TensorFlow GPU（详细教程+错误解决）

1 关于更新源

卖家提供的资料不推荐换源，原因： Jetson TX2 NX 采用的是 aarch64 架构的 Ubuntu 18.04.2 LTS 系统，与 AMD 架构的 Ubuntu 系统不同，而没有找到完美的国内源，所以不推荐大家换源。

我没有换源，但是安装一些库的速度还可以，首先执行以下操作

sudo apt-get update

如图所示：

然后，执行下面的命令

sudo apt-get full-upgrade

如图所示：

命令执行过程，弹出该窗口，我直接选择No，继续：

可以输入命令，查看CUDA

nvcc -V

如图所示：

检查Opencv，首先输入命令

python

然后输入如下python代码

import cv2
cv2.__version__

可以看到Opencv版本号，然后输入退出命令

exit()

如图所示：

测试板载摄像头，写入命令

nvgstcapture-1.0

如图所示：

直接输入q，回车即可关闭摄像头：

2 Jetson TX2 NX 安装 TensorFlow GPU

安装pip命令如下

sudo apt-get install python3-pip python3-dev

可能因为网络原因安装失败，多试几次，如图所示：

升级pip，输入命令

python3 -m pip install --upgrade pip

如图所示：

安装numpy包，输入命令

sudo apt-get install python3-numpy

如图所示：

安装scipy包，输入命令

sudo apt-get install python3-scipy

如图所示：

安装pandas包，输入命令

sudo apt-get install python3-pandas

如图所示：

安装matplotlib包，输入命令

sudo apt-get install python3-matplotlib

如图所示：

安装sklearn包，输入命令

sudo apt-get install python3-sklearn

如图所示：

安装libhdf5-serial-dev包，输入命令

sudo apt-get install libhdf5-serial-dev

如图所示：

安装hdf5-tools包 ，输入命令

sudo apt-get install hdf5-tools

如图所示：

安装libhdf5-dev包 ，输入命令

sudo apt-get install libhdf5-dev

如图所示：

安装zlib1g-dev包 ，输入命令

sudo apt-get install zlib1g-dev

如图所示：

安装zip包 ，输入命令

sudo apt-get install zip

如图所示：

安装libjpeg8-dev包，输入命令

sudo apt-get install libjpeg8-dev

如图所示：

安装liblapack-dev包，输入命令

sudo apt-get install liblapack-dev

如图所示：

安装libblas-dev包，输入命令

sudo apt-get install libblas-dev

如图所示：

安装gfortran包，输入命令

sudo apt-get install gfortran

如图所示：

使用离线方式安装TensorFlow GPU，直接使用U盘将安装文件拷贝到Jetson TX2 NX，然后执行安装命令

pip3 install tensorflow-2.1.0+nv20.4-cp36-cp36m-linux_aarch64.whl

可能出现如下错误，报错原因是下载一些包时，连接中断：

解决办法：由于安装过程会安装一些软件包，为了提高下载速度和稳定性，现在给pip换源

pip永久换源，输入如下命令，创建.pip

mkdir ~/.pip

然后创建pip.conf文件，输入命令

vim ~/.pip/pip.conf

进入之后，按i进入编辑模式，复制下面的清华源

[global]
index-url = https://pypi.tuna.tsinghua.edu.cn/simple
trusted-host = pypi.tuna.tsinghua.edu.cn

然后按ESC退出，按:wq保存退出

然后再执行安装命令，如图所示：

安装h5py包时报错，如图所示：

尝试网上很多解决办法，都无效，最终通过输入下列命令安装，成功安装h5py包

sudo apt install python3-h5py

如图所示：

Keras 是一个用 Python 编写的高级神经网络 API，安装命令如下

sudo pip3 install keras

如图所示：

希望本文对大家有帮助，上文若有不妥之处，欢迎指正

分享决定高度，学习拉开差距