文章目录
前言
随着人工智能的不断发展,机器学习这门技术也越来越重要,在嵌入式图像识别中我们难免会使用到jetson nano做图像处理,本章讲述怎么样在jetson nano上安装pytorch和pytorchvision两个重要的东西
安装需要的下载文件
下载链接:https://pan.baidu.com/s/1bQGfYtmOB-EaqvKnnJZE_w
提取码:lpeh
一、Jetson nano 的镜像系统
在英伟达官方网站可以找到烧写镜像的具体步骤,这里不在赘述,镜像烧写按照步骤一步一步来就可以了,内存卡推荐使用32gb以上的,镜像烧写完成后进入系统
二、修改镜像源
1.在终端里使用vim /etc/apt/sources.list将原来的内容注释掉换成中科大源(清华源下载比较慢)
# 默认注释了源码镜像以提高 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
2.系统更新:
sudo apt-get update
sudo apt-get upgrade -y
三.pytorch安装
1.首先安装pip和Cython
sudo apt-get install python3-pip libopenblas-base libopenmpi-dev
pip3 install Cython
Cython如果下载速度慢可以在命令后面加临时软件源 -i https://pypi.mirrors.ustc.edu.cn/simple
pip3 install Cython -i https://pypi.mirrors.ustc.edu.cn/simple
下面开始安装pytorch,安装pytorch是可以用conda安装的,不过经过多次安装发现,conda需要配置虚拟环境有时python的版本设置可能不对,conda安装pytorch还是相当麻烦的,这里采用pytorch的离线包安装(下载链接前言中附赠)。
下载好文件对文件解压进入文件夹中
jetbot@jetbot-desktop:~/Downloads$ ls
pytorch-torchvison.zip
jetbot@jetbot-desktop:~/Downloads$ unzip pytorch-torchvison.zip
jetbot@jetbot-desktop:~/Downloads$ ls
pytorch-torchvison pytorch-torchvison.zip
jetbot@jetbot-desktop:~/Downloads$ cd pytorch-torchvison/
jetbot@jetbot-desktop:~/Downloads/pytorch-torchvison$ ls
torch-1.8.0-cp36-cp36m-linux_aarch64.whl torchvision
进入文件夹看到这俩,接着开始安装pytorch
sudo pip3 install torch-1.6.0-cp36-cp36m-linux_aarch64.whl
等待安装结束进入python3中import验证一下pytorch的导包
四.安装pytorchvision
还是在当前目录下,进入torchvision文件夹里,在终端里键入以下命令对pytorchvision安装
sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libavcodec-dev libavformat-dev libswscale-dev
sudo python3 setup.py install
等待pyutorchvision安装成功进入python3中import导包进行验证。