Jetson Nano新机安装Tensorflow(版本JetPack4.4)

记录安装Tensorflow的一些坑!


前言

最近实验室新进一批英伟达的设备,因实验需要,记录一下nano新机从烧录到安装Tensorflow的全过程


一、格式化and烧录

1、格式化SD卡。格式化的软件为SD Card Formatter: SD Memory Card Formatter for Windows/Mac | SD Association

 注意:win11可能会识别不到读卡器里的磁盘,建议用win10

2、烧录镜像。

首先去官网下载所需要的镜像文件,这里我们用的是nano jetpack4.4:JetPack SDK 4.4 archive | NVIDIA Developer

下载完成以后,使用烧录软件进行烧录:https://www.balena.io/etcher/

 烧录完成之后直接插入到嵌入式平台上的SD卡位置,然后连接所有的线然后开机。

二、环境配置

1.修改环境变量

Jetson Nano利用官方镜像进行安装后,系统已经安装好了JetPack,cuda,cudaa,OpenCV等组件,需要修改下环境变量才可以使用。

利用gedit打开 ~ 路径下.bashrc文件:

sudo gedit ~./bashrc

文件的最后添加以下三行:

export PATH=/usr/local/cuda-10.2/bin:$PATH

export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH

export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-10.2

重新执行.bashrc文件,直接生效;

source ~./bashrc

输入nvcc -V命令进行测试,如果显示如下信息,证明修改正确。

nvcc -V

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Wed_Oct_23_21:14:42_PDT_2019
Cuda compilation tools, release 10.2, V10.2.89

2.更换国内镜像源

NVIDIA官方提供的Linux镜像版本为Ubuntu 18.04 LTS,镜像默认的是Ubuntu官方源,在国内使用该源下载程序时速度较慢,所以需要更换。

首先,打开/etc/apt/sources.list文件,注释原内容

sudo gedit /etc/apt/sources.list

 在末尾添加下述内容(以清华大学镜像源为例,注意镜像源需要支持arm64架构)

# 默认注释了源码镜像以提高 apt update 速度,如有需要可自行取消注释
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ bionic main restricted universe multiverse
# deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ bionic main restricted universe multiverse
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ bionic-updates main restricted universe multiverse
# deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ bionic-updates main restricted universe multiverse
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ bionic-backports main restricted universe multiverse
# deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ bionic-backports main restricted universe multiverse
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ bionic-security main restricted universe multiverse
# deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ bionic-security main restricted universe multiverse
 
# 预发布软件源,不建议启用
# deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ bionic-proposed main restricted universe multiverse
# deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu-ports/ bionic-proposed main restricted universe multiverse

保存并关闭文件,然后输入以下命令来更新源:

sudo apt-get update

其它部分常用镜像源:

中科大镜像源:

# 默认注释了源码镜像以提高 apt update 速度,如有需要可自行取消注释
deb http://mirrors.ustc.edu.cn/ubuntu-ports bionic main restricted universe multiverse
# deb-src http://mirrors.ustc.edu.cn/ubuntu-ports bionic main restricted universe multiverse
deb http://mirrors.ustc.edu.cn/ubuntu-ports bionic-updates main restricted universe multiverse
# deb-src http://mirrors.ustc.edu.cn/ubuntu-ports bionic-updates main restricted universe multiverse
deb http://mirrors.ustc.edu.cn/ubuntu-ports bionic-backports main restricted universe multiverse
# deb-src 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 universe multiverse
# deb-src http://mirrors.ustc.edu.cn/ubuntu-ports bionic-security main restricted universe multiverse
 
# 预发布软件源,不建议启用
# deb http://mirrors.ustc.edu.cn/ubuntu-ports bionic-proposed main restricted universe multiverse
# deb-src http://mirrors.ustc.edu.cn/ubuntu-ports bionic-proposed main restricted universe multiverse

阿里源:

# 默认注释了源码镜像以提高 apt update 速度,如有需要可自行取消注释
deb https://mirrors.aliyun.com/ubuntu-ports/ bionic main restricted universe multiverse
# deb-src https://mirrors.aliyun.com/ubuntu-ports/ bionic main restricted universe multiverse
deb https://mirrors.aliyun.com/ubuntu-ports/ bionic-updates main restricted universe multiverse
# deb-src https://mirrors.aliyun.com/ubuntu-ports/ bionic-updates main restricted universe multiverse
deb https://mirrors.aliyun.com/ubuntu-ports/ bionic-backports main restricted universe multiverse
# deb-src https://mirrors.aliyun.com/ubuntu-ports/s bionic-backports main restricted universe multiverse
deb https://mirrors.aliyun.com/ubuntu-ports/ bionic-security main restricted universe multiverse
# deb-src https://mirrors.aliyun.com/ubuntu-ports/ bionic-security main restricted universe multiverse
 
# 预发布软件源,不建议启用
# deb https://mirrors.aliyun.com/ubuntu-ports/ bionic-proposed main restricted universe multiverse
# deb-src https://mirrors.aliyun.com/ubuntu-ports/ bionic-proposed main restricted universe multiverse

三、pip安装

python3

sudo apt-get install python3-pip python3-dev

 更新pip,保证是最新版本

pip3 install --upgrade pip

 四、Jtop安装

1、依赖安装

sudo apt-get install git cmake
sudo apt-get install python3-dev
sudo apt-get install libhdf5-serial-dev hdf5-tools
sudo apt-get install libatlas-base-dev gfortran

2、安装jtop

sudo -H pip3 install -U jetson-stats

sudo reboot重启后,在终端输入jtop可打开jtop,控制风扇、查看信息等。

如果不行,需要激活jtop服务再重启

 sudo systemctl restart jetson_stats.service

 五、安装Tensorflow

踩了很多坑,这里一一注明。

1、安装一些必要的依赖库

安装一些学习包。

sudo apt install python3-scipy -y
sudo apt install python3-pandas -y
sudo apt install python3-sklearn -y
sudo apt install python3-seaborn -y

下面这些库可以一起安装,不会报错。

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

2、安装numpy

需要先升级 protobuf 版本至 3.3.0:

sudo pip3 install protobuf==3.3.0

安装 numpy:

sudo pip3 install -U numpy==1.16.1

注意这里numpy的版本,太高或者太低多多少少都会导致一些后面安装的问题,如果不指定版本,默认安装1.19.5版本会导致后面装h5py的时候报非法指令的错误。

3、安装h5py包

这个包对依赖的版本要求真的非常高!出了各种各样的问题,最终选择下载源文件手动安装

笔记本在清华的镜像软件源下载h5py==2.10.0(h5py-2.10.0.tar.gz)Links for h5py (tsinghua.edu.cn)

下载完成用U盘拷贝到nano中

在nano里面解压文件,并在对应文件夹中(文件夹可以看到setup.py)打开终端

注意!在安装前先要安装Cython==3.0.0a10!如果有Cython也要保证版本一致!

sudo pip3 install -U Cython==3.0.0a10

接着运行setup.py来装h5py,注意这里的python3不能少,否则系统默认是2.7的版本,会找不到Cython

sudo python3 ./setup.py install

一段漫长的编译以后,h5py==2.10.0就装好了!

4、安装keras-applications包

这个包要在安装完 h5py 包之后才能正常安装。

sudo pip3 install -U keras-applications

5、安装future包

sudo pip3 install -U future

6、安装 setuptools 和 testresources 包

sudo pip3 install -U setuptools testresources

7、检查相应库的版本

最后检查一遍相应库的版本看是否满足安装 Tensorflow 的条件。

检查相应库的版本:

pip3 list

安装 Tensorflow 的条件:

numpy==1.16.1, future==0.17.1, mock==3.0.5, h5py==2.9.0, gast==0.2.2, keras_preprocessing==1.0.5, keras_applications==1.0.8, scipy==1.4.1

8、安装Tensorflow

使用pip3安装TensorFlow,该命令将安装与JetPack 4.4兼容的TensorFlow的最新版本:

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

如果速度太慢可以pip暂时换源:(在上面命令后加 -i https://pypi.tuna.tsinghua.edu.cn/simple)

sudo pip3 install --pre --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v44 tensorflow -i https://pypi.tuna.tsinghua.edu.cn/simple

等待一段时间后,Tensorflow就装好啦!!

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值