jetson orin nano学习(torch+OpenCV+yolov5+)

一:入门第一件事:跟着商家教程配置哈哈

指令:nvidia-smi -h 帮助命令

sudo jtop --查看nvidia的gpu状态

Tip:教程下载的pytorth,cuda,cudnn版本不一定是你项目符合的,要提前想好

1.2 安装虚拟环境包(要安装minifoge3,因为nvdia设备架构不能按照conda和miniconda)

安装过程可以查看:
【nano系列】jetson nano 安装conda管理环境(三)_jetson 安装conda-CSDN博客

https://github.com/conda-forge/miniforge

二,部署一下yolov5练练手(tensor 加速版)

2.1涉及网站汇总:

1.pytorch安装

https://forums.developer.nvidia.com/t/pytorch-for-jetson/72048

2.部署过程

Jetson Nano部署YOLOv5与Tensorrtx加速------(自己走一遍全过程记录)_jetson nano yolov5-CSDN博客

2.2安装pytorch和torchvision

#tip:安装时一定要明确自己的python版本和jetback版本,torch只有在nvdia官网下载才能适配。

(1)在2.1中第一个网站找到要下载的pytorch安装包,下载

(2)在其页面下方打开installsion查看下载的pytorch需要的torchvision对应版本

(3)安装Torchvision的步骤

**(1)安装依赖库:**在编译Torchvision前,需安装以下依赖:

复制代码
sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libavcodec-dev libavformat-dev libswscale-dev

(2)克隆指定版本的Torchvision源码

python 复制代码
git clone --branch v0.15.1 https://github.com/pytorch/vision torchvision
cd torchvision

(版本号0.15.1根据对应关系调整)

(3)设置环境变量并编译安装

python 复制代码
cd torchvision#进入目录
export BUILD_VERSION=0.15.1  # 与Torchvision版本一致
python3 setup.py install --user
复制代码
大功告成!!

2.3,安装opencv(不是一般的opencv,要符合nvdia配置)

安装后面编译需要用到的依赖库

python 复制代码
sudo apt install -y build-essential checkinstall cmake pkg-config yasm git gfortran
sudo apt update
sudo apt install -y libgstreamer1.0-dev libgstreamer-plugins-base1.0-dev
sudo apt install -y libjpeg8-dev libjasper-dev libpng12-dev libtiff5-dev libavcodec-dev libavformat-dev libswscale-dev libdc1394-22-dev libxine2-dev libv4l-dev
sudo apt install -y libgstreamer1.0-dev libgstreamer-plugins-base1.0-dev libgtk2.0-dev libtbb-dev libatlas-base-dev libfaac-dev libmp3lame-dev libtheora-dev libvorbis-dev libxvidcore-dev libopencore-amrnb-dev libopencore-amrwb-dev x264 v4l-utils
sudo apt install python-dev python-numpy libtbb2 libtbb-dev libjpeg-dev libpng-dev libtiff-dev libjasper-dev libdc1394-22-dev

下载两个包

Tags · opencv/opencv

Tags · opencv/opencv_contrib

将contrib解压在opencv文件夹下面

进行以下步骤

python 复制代码
cd opencv-4.5.5
mkdir build

cmake \
-DCMAKE_BUILD_TYPE=Release \
-DCMAKE_INSTALL_PREFIX=/usr/local \
-DOPENCV_ENABLE_NONFREE=1 \
-DBUILD_opencv_python2=1 \
-DBUILD_opencv_python3=1 \
-DWITH_FFMPEG=1 \
-DCUDA_TOOLKIT_ROOT_DIR=/usr/local/cuda \
-DCUDA_ARCH_BIN=7.2 \
-DCUDA_ARCH_PTX=7.2 \
-DWITH_CUDA=1 \
-DENABLE_FAST_MATH=1 \
-DCUDA_FAST_MATH=1 \
-DWITH_CUBLAS=1 \
-DOPENCV_GENERATE_PKGCONFIG=1 \
-DOPENCV_EXTRA_MODULES_PATH=../opencv_contrib-4.5.4/modules \
..


make -j6
sudo make install

但import cv2时,还是会报no module "cv2"的报错,到build目录下,找到python_load文件夹,这个文件夹就是cv2编译的python package

python 复制代码
cd opencv-4.5.5/build/python_loader
sudo pip3 install .

再测试一下,发现和jtop看到的是一致的

详细配置可以去看up主:荔枝澄的个人空间-荔枝澄个人主页-哔哩哔哩视频

2.4配置nvcc -V

2.4.1 打开终端

python 复制代码
#打开终端,输入命令
vi .bashrc

拉到最后,在最后添加这些 (10.2根据自己版本改)

python 复制代码
export PATH=/usr/local/cuda-10.2/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
export CUDA_ROOT=/usr/local/cuda

应用当前配置(刷新一下)

python 复制代码
source ~/.bashrc

2.4.2 安装后输出命令查看

python 复制代码
nvcc -V

今天就学到这啦,注意,有些步骤编译要很久,不是出错了哦!

++谢谢++

相关推荐
The_cute_cat2 分钟前
JavaScript的初步学习
开发语言·javascript·学习
Binary_ey1 小时前
超表面重构卡塞格林望远镜 | 从传统架构到新型光学系统
学习·软件需求·光学软件·超表面
roman_日积跬步-终至千里2 小时前
【学习线路】机器学习线路概述与内容关键点说明
人工智能·学习·机器学习
天水幼麟3 小时前
python学习笔记(深度学习)
笔记·python·学习
you45804 小时前
小程序学习笔记:使用 MobX 实现全局数据共享,实例创建、计算属性与 Actions 方法
笔记·学习·小程序
Brookty5 小时前
【MySQL】JDBC编程
java·数据库·后端·学习·mysql·jdbc
DKPT5 小时前
Java设计模式之结构型模式(外观模式)介绍与说明
java·开发语言·笔记·学习·设计模式
编程小白gogogo5 小时前
Spring学习笔记
笔记·学习·spring
qq_527887875 小时前
【学习笔记】Python中主函数调用的方式
笔记·学习
Chef_Chen6 小时前
从0开始学习R语言--Day37--CMH检验
学习