Bootstrap

【玩转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

如图所示:

在这里插入图片描述
希望本文对大家有帮助,上文若有不妥之处,欢迎指正

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

;