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

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

++谢谢++

相关推荐
好学且牛逼的马17 分钟前
#13 学习反馈
java·学习·ai编程
蓝婷儿1 小时前
6个月Python学习计划 Day 8 - Python 函数基础
开发语言·python·学习
mit6.8241 小时前
[Rust_1] 环境配置 | vs golang | 程序运行 | 包管理
开发语言·学习·rust
guts3503 小时前
【pytorch学习】土堆pytorch学习笔记2
pytorch·笔记·学习
吃着火锅x唱着歌3 小时前
PHP7内核剖析 学习笔记 第九章 PHP基础语法的实现
笔记·学习·php
缘友一世3 小时前
LangFuse:开源LLM工程平台的革新实践
学习·开源·agent
aa32422432433 小时前
Scratch节日 | 粽子收集
学习·青少年编程·节日
white.tie4 小时前
Pandas学习入门一
学习·pandas
jz_ddk6 小时前
[学习] C语言多维指针探讨(代码示例)
linux·c语言·开发语言·学习·算法
艾莉丝努力练剑10 小时前
深入详解编译与链接:翻译环境和运行环境,翻译环境:预编译+编译+汇编+链接,运行环境
c语言·开发语言·汇编·学习