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

您可以通过打赏支持我的工作,谢谢!

  • 5
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 4
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值