【剁手系列】Jetson Nano（二）安装TensorFlow GPU以及一些常用包

最新推荐文章于 2021-01-06 01:36:06 发布

destiny1679

最新推荐文章于 2021-01-06 01:36:06 发布

阅读量1.3k

点赞数

分类专栏： Jetson nano 文章标签： jetson nano TensorFlow

本文链接：https://blog.youkuaiyun.com/destiny35456/article/details/103308903

版权

Jetson nano 专栏收录该内容

6 篇文章

订阅专栏

本指南详细介绍了在Jetson Nano上安装TensorFlow GPU的步骤，包括安装pip、依赖包、TensorFlow以及常用包的过程。通过使用清华源加速下载，确保了安装的效率。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在安装TensorFlow GPU之前，需要先安装一些依赖包。

安装pip
因为Jetson Nano中已经安装了Python3.6版本，所以安装pip还是比较简单的
```
sudo apt-get install python3-pip python3-dev
python3 -m pip install --upgrade pip
```
安装后pip后，需要把它升级到最新版，升级后pip版本为19.3.1。这里面升级后会有一个小Bug，需要手动改一下，升级后都得改，其他版本的也是，通病了。
```
sudo vim /usr/bin/pip3
```
将原来的
```
from pip import main
if __name__ == '__main__':
    sys.exit(main())
```
改成
```
from pip import __main__
	if __name__ == '__main__':
	    sys.exit(__main__._main())
```
修改结束后保存。运行pip3 -V成功后显示

安装依赖包

sudo apt-get install libhdf5-serial-dev hdf5-tools

安装TensorFlow
我们现在英伟达官网上下载包，不然安装要老半天了，下面是链接，建议用迅雷，只要速度快，用什么都行。
tf_gpu-1.14.0+nv19.10-py3
然后执行命令安装，至于加个清华源是为了防止缺某些依赖包，需要下载安装，如果用官方源的话慢，用上清华源可以快很多。
```
sudo pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow_gpu-1.14.0+nv19.10-cp36-cp36m-linux_aarch64.whl
```
安装完成后，随便找个代码试试，没错就Ok了。

常用包安装下载
建议一个一个分开来安装，不然容易卡死，我为了方便打，就放在一起了。

sudo pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple Cython numpy scipy scikit-image matplotlib ipython h5py leveldb networkx nose pandas python-dateutil protobuf python-gflags pyyaml Pillow six keras