Nano环境下配置+torch&tensorflow安装

CUDA与Torch、Torchvision版本对应

CUDA下载链接

JetPack4.5下载链接

recommend:

CUDA	Torch	Torchvision
9.0/10.0	1.1.0	0.3.0
9.2/10.0	1.2.0	0.4.0
–	1.3.0	0.4.2
9.2/10.1	1.4.0	0.5.0
9.2/10.1/10.2	1.5.0	0.6.0
9.2/10.1/10.2	1.6.0	0.7.0
9.2/10.1/10.2/11.0	1.7.0	0.8.1
10.2/11.0	1.8.0	0.9.0

1 配置Torch

pip install torch===1.5.0 torchvision===0.6.0 -f https://download.pytorch.org/whl/torch_stable.html

2 配置CUDA

# 安装环境变量：
vim ~/.bashrc   #或者采用 source gedit ~/.bashrc
 
#在最后写入并保存：
export CUDA_HOME=/usr/local/cuda
export PATH=$PATH:$CUDA_HOME/bin
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
 
#使其生效：
source ~/.bashrc
 
#查看是否生效：
cat /proc/driver/nvidia/version
nvcc -V

3 Jestin nano 刷机后环境初始化

分别执行以下命令，即可查看自己的jetson nano 预搭载的CUDA版本
sudo pip3 install jetson-stats
sudo jtop

1.镜像下载

（1）CUDA10.0的镜像：https://developer.nvidia.com/embedded/dlc/jetson-nano-dev-kit-sd-card-image

（2）CUDA10.2的镜像：https://developer.nvidia.com/jetson-nano-sd-card-image

附：官网详细介绍

2.镜像文件写入microSD卡

https://www.balena.io/etcher/

3.ubuntu流程（略）

4.开机后环境配置

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

sudo apt-get install libhdf5-dev
sudo apt-get install python-h5py

5.Tensorflow and pytorch安装

tensorflow 快速安装链接：https://developer.download.nvidia.com/compute/redist/jp/v42/tensorflow-gpu/
Tensorflow for Jetson Nano
pytorch快速安装链接:官网whl链接

#install tensorflow
sudo pip install --ignore-installed --upgrade tensorflow_gpu-1.3.0-cp36-cp36m-linux_x86_64.whl

安装torch

wget https://nvidia.box.com/shared/static/p57jwntv436lfrd78inwl7iml6p13fzh.whl -O torch-1.8.0-cp36-cp36m-linux_aarch64.whl
sudo apt-get install python3-pip libopenblas-base libopenmpi-dev 
pip3 install Cython
pip3 install numpy torch-1.8.0-cp36-cp36m-linux_aarch64.whl

安装torchvision （官网github源码下载）

$ sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libavcodec-dev libavformat-dev libswscale-dev
$ git clone --branch <version> https://github.com/pytorch/vision torchvision   # see below for version of torchvision to download
$ cd torchvision
$ export BUILD_VERSION=0.x.0  # where 0.x.0 is the torchvision version  
$ python3 setup.py install --user
$ cd ../  # attempting to load torchvision from build dir will result in import error
$ pip install 'pillow<7' # always needed for Python 2.7, not needed torchvision v0.5.0+ with Python 3.6

#换成码云下载torchvision
$ git clone --branch v0.7.0 https://gitee.com/zero-one-game/vision 

指定镜像安装pillow

sudo apt-get install libjpeg8 libjpeg62-dev libfreetype6 libfreetype6-dev
python3 -m pip install -i https://mirrors.aliyun.com/pypi/simple pillow

6. archiconda3:
   https://github.com/yqlbu/archiconda3

#open the file
vim ~/.bashrc
#archiconda3
export PATH="/home/klx/archiconda3/bin:$PATH"
#update the file
source ~/.bashrc

enable all Ubuntu packages:
$ sudo apt-add-repository universe
$ sudo apt-add-repository multiverse
$ sudo apt-add-repository restricted


add ROS repository to apt sources
$ sudo sh -c '. /etc/lsb-release && echo "deb http://mirrors.ustc.edu.cn/ros/ubuntu/ $DISTRIB_CODENAME main" > /etc/apt/sources.list.d/ros-latest.list'
$ sudo apt-key adv --keyserver 'hkp://keyserver.ubuntu.com:80' --recv-key C1CF6E31E6BADE8868B172B4F42ED6FBAB17C654


install ROS Base
$ sudo apt-get update
$ sudo apt-get install ros-melodic-ros-base


add ROS paths to environment
sudo sh -c 'echo "source /opt/ros/melodic/setup.bash" >> ~/.bashrc'

安装opencv

安装opencv依赖项，用aptitude来进行操作：
# Update
sudo apt-get update
sudo apt-get upgrade
# Pre-requisites
sudo aptitude install build-essential cmake unzip pkg-config
sudo aptitude  install libjpeg-dev libpng-dev libtiff-dev
sudo aptitude  install libavcodec-dev libavformat-dev libswscale-dev libv4l-dev
sudo aptitude  install libxvidcore-dev libx264-dev
 
#下面这个会告诉你有冲突项，你第一次选择n，第二次之后选择y即可
sudo aptitude  install libgtk-3-dev
sudo aptitude  install libatlas-base-dev gfortran
sudo aptitude  install python3-dev

1.首先扩大jetson nano的内存，因为要下载opencv的源码并编译，之后还要安装tensorflow，所以可能内存不够大，要增加内存。可以进入github：https://github.com/JetsonHacksNano查看并下载
或者直接在命令行操作

git clone https://github.com/JetsonHacksNano/installSwapfile
./installSwapfile.sh #编译文件，默认扩大6G内存

2.下载编译opencv

git clone https://github.com/JetsonHacksNano/buildOpenCV
./buildOpenCV.sh #编译opencv

或者 pip3 install opencv-python
3.测试

python3
import cv2
cv2.__version__

4. The second method: sudo apt-get install python3-opencv
5. The third method:https://www.jianshu.com/p/99bdc2472423

Jetson Nano使用教程：
（1）http://www.yoyojacky.com/?p=513

4 Jetson Nano 更换镜像源

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

打开/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

2、其他镜像源
下面提供了国内几个支持arm64架构的Ubuntu镜像源：

清华大学镜像源：

# 默认注释了源码镜像以提高 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


中国科学技术大学镜像源：

# 默认注释了源码镜像以提高 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

3、系统更新
系统源和软件源配置完成以后，使用下述两条命令更新：

sudo apt update
sudo apt upgrade -y

附件

Jetson Nano 版本对应

JetPack与Pytorch版本

PyTorch与TorchVision版本