Nano环境下配置+torch&tensorflow安装
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卡
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
- 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__
- The second method:
sudo apt-get install python3-opencv
- 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