JetSon Nano开发板安装tensorflow-gpu

方案一：

安装相关依赖：

$ sudo apt-get install libhdf5-serial-dev hdf5-tools
$ sudo apt-get install python3-pip
$ pip3 install -U pip
$ sudo apt-get install zlib1g-dev zip libjpeg8-dev libhdf5-dev 
$ sudo python3 -m pip install -U numpy grpcio absl-py py-cpuinfo psutil portpicker grpcio six mock requests gast h5py astor termcolor

安装TensorFlow：

$ pip3 install --pre --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v42 tensorflow-gpu

 如上面安装出错，尝试更换不同的版本，自己指定版本如下：

$ pip3 install --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v42 tensorflow-gpu==1.13.0+nv19.1

 

方案二：

更换apt源，Jetson设备默认的apt镜像源在海外，速度很慢，这里用一个国内的源代替海外源。这里用的是中科大的源，内容如下：

deb http://mirrors.ustc.edu.cn/ubuntu-ports/ bionic-updates main restricted
deb http://mirrors.ustc.edu.cn/ubuntu-ports/ bionic universe
deb http://mirrors.ustc.edu.cn/ubuntu-ports/ bionic-updates universe
deb http://mirrors.ustc.edu.cn/ubuntu-ports/ bionic multiverse
deb http://mirrors.ustc.edu.cn/ubuntu-ports/ bionic-updates multiverse

deb http://mirrors.ustc.edu.cn/ubuntu-ports/ bionic-backports main restricted universe multiverse

deb http://mirrors.ustc.edu.cn/ubuntu-ports/ bionic-security main restricted
deb http://mirrors.ustc.edu.cn/ubuntu-ports/ bionic-security universe
deb http://mirrors.ustc.edu.cn/ubuntu-ports/ bionic-security multiverse

更换apt源，如果su root失败，输入sudo passwd,按提示设置一个密码，确认后再su root即可，因为默认下没有初始化root。

su root #切换到root用户，不然没有更改source.list的权限
sudo mv /etc/apt/sources.list /etc/apt/sources.list.bak # 将原镜像源文件做一个备份
sudo vi /etc/apt/sources.list  #用文本编辑器打开源文件进行编辑，将上述中科大的apt源内容写入这个文件中。

使apt源生效：

sudo apt-get update
sudo apt-get upgrade

安装需要的工具

sudo apt install htop screen nano

简单测试，Jetson-nano的OS镜像已经自带了JetPack，cuda，cudnn，opencv等都已经安装好，不再需要额外的配置，这里简单测试一下。系统里面自带了cudnn的测试程序，在/usr/src/cudnnsamplesv7/目录下，这里用mnistCUDNN程序进行测试。

cp -rvf /usr/src/cudnn_samples_v7/mnistCUDNN ~  #将mnistCUDNN复制到home目录
cd ~/mnistCUDNN
make  #编译源代码
chmod a+x mnistCUDNN  #为可执行文件添加执行权限
./mnistCUDNN  #执行

如果没报错则测试成功。

安装相关的依赖库和工具

sudo apt-get install libhdf5-serial-dev hdf5-tools zlib1g-dev zip libjpeg8-dev libhdf5-
dev  python3-pip

系统里面默认是没有python3-pip的，所以需要自己安装。

安装相关的python库

pip3 install -U numpy  # 需要编译安装，用时很长，所以单独安装
pip3 install -U h5py   # 需要编译安装，用时非常长，我的板子装这个20多分钟才装完
pip3 install -U grpcio absl-py py-cpuinfo psutil portpicker six mock requests gast astor termcolor

安装tensorflow

pip3 install --pre --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v42 tensorflow-gpu

如果安装tensorflow报错，可尝试更换不同版本，如下：

$ pip3 install --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v42 tensorflow-gpu==1.13.0+nv19.1

 

另外Jetson nano安装matplotlib：

 sudo apt-get install python3-matplotlib

 

 

注意：

如果由于pip3升级后出现importerror:cannot import name main问题，

su root获取root权限，然后打开vi /usr/bin/pip3

将下面三行：

from pip import main
if __name__ == '__main__':
    sys.exit(main())

改为下面三行即可：

from pip import __main__
if __name__ == '__main__':
    sys.exit(__main__._main())

 参考：Nano入门教程软件篇-安装TensorFlow-GPU

Jetson-Nano开箱配置及Tensorflow安装使用 简介