Jetson Nano安装TensorFlow和PyTorch【第四部分】

一、写在前面

 

第四部分承接第三部分，这个部分不是必要的，请根据自己的需要选择是否安装两个流行的深度学习框架。

Jetson系列可以说是高端的边缘AI设备，十分适合用于深度学习。然而，NVIDIA通过Jetpack提供的OS和SDK只包括CUDA、cuDNN、OpenCV三个常用的库，没有包含TensorFlow和PyTorch的CPU和GPU版本。

本部分介绍virtualenv、tensorflow-gpu、pytorch-gpu的安装，并对它们的安装情况进行验证。

二、准备工作

在Nano USB系统中，由于系统下载源均为Ubuntu和NVIDIA提供，在国内网络环境会导致一些库或包下载缓慢。所以，需要在Nano USB系统中添加清华镜像源。

（1）添加清华镜像源

打开终端（Terminal），输入以下代码。

sudo gedit /etc/apt/sources.list

复制以下内容到文件的最前面。

deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic main restricted universe multiverse
# deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic main restricted universe multiverse
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-updates main restricted universe multiverse
# deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-updates main restricted universe multiverse
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-backports main restricted universe multiverse
# deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-backports main restricted universe multiverse
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-security main restricted universe multiverse
# deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-security main restricted universe multiverse

保存并退出，返回到终端（Terminal）。

（2）更新下载源和软件

输入以下代码，更新下载源和已安装软件。

sudo apt update
sudo apt upgrade

三、安装TensorFlow和PyTorch

09、安装virtualenv

virtualenv是一个创建python虚拟环境的软件。

打开终端（Terminal），输入以下代码：

sudo apt-get install virtualenv

10、安装Tensorflow GPU版本

（1）创建tensorflow虚拟环境

python3 -m virtualenv -p python3 tensorflow

（2）安装tensorflow gpu

NVIDIA官方提供了安装指引，详见网站：Installing TensorFlow for Jetson Platform :: NVIDIA Deep Learning Frameworks Documentation

因国内网络问题，本教程提供百度网盘下载地址：

【Jetpack v461/462专用安装包，含tensorflow & pytorch】

链接：https://pan.baidu.com/s/1EaO8mp7cn0tHpzRU-6hcTw?pwd=ouhx 
提取码：ouhx 
（3）安装时可能出现依赖问题，如果自己无法解决，可以下载本教程提供的镜像。

11、安装Pytorch GPU版本

（1）创建pytorch虚拟环境

python3 -m virtualenv -p python3 pytorch

（2）安装pytorch gpu

NVIDIA官方提供了安装指引，详见网站：

Installing PyTorch for Jetson Platform :: NVIDIA Deep Learning Frameworks Documentation

因国内网络问题，本教程提供百度网盘下载地址：前面已经提供。

（3）安装时可能出现依赖问题，如果自己无法解决，可以下载本教程提供的镜像。

 作者已经将U盘文件保存为.pmf模式，可以使用DiskGenius还原分区。

插入ext4格式的U盘，打开DiskGenius，选择“从镜像文件还原分区”，选择下载好的“whole_1128.pmf”（压缩包大小为GB），点击开始。整个过程在15分钟内完成，取决于电脑配置。

12、验证所有安装

注意，tensorflow和pytorch都要安装NVIDIA官方提供的Jetson平台专用版本，否则容易出问题。

按照本教程安装完tensorflow和pytorch后，使用以下代码验证安装情况。

（1）验证tensorflow

# 激活tensorflow虚拟环境

source tensorflow/bin/activate

python

import tensorflow

# 若通过，则允许用户输入下一行Python代码

exit()

（2）验证pytorch

# 激活pytorch虚拟环境

source pytorch/bin/activate

python

import torch

# 若通过，则允许用户输入下一行Python代码

exit()

三、写在后面

本教程共分为四个部分，各个部分如下：

	名称	对应步骤
第一部分	环境准备	01、下载VMware 02、下载虚拟机文件（评论或私聊作者，获取解压密码）
第二部分	安装系统	03、Nano EMMC 刷系统 04、Nano EMMC 开机 05、Nano EMMC 安装SDK
第三部分	U盘启动	06、复制系统到U盘（评论或私聊作者，获取解压密码） 07、修改启动配置文件 08、启动Nano USB系统
第四部分	框架安装	09、安装virtualenv 10、安装Tensorflow GPU 11、安装Pytorch GPU 12、验证所有安装（评论或私聊作者，获取解压密码）

下载链接和解压密码见

Jetson Nano系统安装之环境准备【第一部分】http://t.csdn.cn/GULNw

四、参考网站（NVIDIA）

generic-no-api_r2

NVIDIA SDK Manager | NVIDIA Developer

Installing TensorFlow for Jetson Platform :: NVIDIA Deep Learning Frameworks Documentation

Installing PyTorch for Jetson Platform :: NVIDIA Deep Learning Frameworks Documentation

您可以通过打赏支持我的工作，谢谢！