TX2之tensorflow环境部署安装

时间:2018-12-25
本文章向大家介绍TX2之tensorflow环境部署安装,主要包括TX2之tensorflow环境部署安装使用实例、应用技巧、基本知识点总结和需要注意事项,具有一定的参考价值,需要的朋友可以参考一下。

最近nvidia官网发布了专门针对tx2的tensorflow-gpu安装包,这样将TX2上部署tensorflow的难度大大降低,只需几个步骤即可。

1 刷机jetpack3.3

首先TX2必须输3.3版本的jetpack,因为截止目前nvidia发布的tensorflow只支持3.3版本的jetpack,如图所示,日后如果有更新了可以刷其它版本的,所以第一步就是给TX2刷机。具体的刷机方法网上有很多,我这里就不再赘述。

2 安装pip3

因为我想安装python3版本的tensorflow,所以需要用pip3来安装,而刷完机后默认是不带有pip和pip3的,因此需要自己装一下

我尝试过使用如下指令安装

sudo apt-get update
sudo apt-get install python3-pip

但是运行时提示我找不到python3-pip的包,网上查方法都说是要更新源,我已经运行过update还不行,因此换一种安装方法

该链接中找到get-pip.py文件,复制下来,保存在TX2本地上,然后运行以下指令即可

sudo python3 get-pip.py

3 安装tensorflow

按照官网教程,只需要运行如下指令

sudo pip3 install --extra-index-url https://developer.download.nvidia.com/compute/redist/jp33 tensorflow-gpu

但是我运行时报错如下:

SystemError: Cannot compile 'Python.h'. Perhaps you need to install python-dev|python-devel

表示缺少一些依赖,解决方法是直接安装缺的依赖

sudo apt-get install python-dev

安装完之后再运行上边sudo pip3 install语句即可,安装过程中要保证网络通畅,否则会报超时等错误,直接重新运行就行了。

4 验证

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

import tensorflow as tf
tf.__version__