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

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

++谢谢++

相关推荐
技术求索者11 分钟前
c++学习
开发语言·c++·学习
FAREWELL000753 小时前
Untiy基础学习(六)MonoBehaviour基类的简单介绍
学习·unity·游戏引擎
Logintern093 小时前
【每天学习一点点】使用Python的pathlib模块分割文件路径
开发语言·python·学习
帅云毅4 小时前
Screeps Arena基础入门
学习·js·印象笔记
缘友一世5 小时前
深度学习系统学习系列【5】之深度学习基础(激活函数&损失函数&超参数)
人工智能·深度学习·学习
虾球xz5 小时前
游戏引擎学习第260天:在性能分析器中实现钻取功能
网络·c++·学习·游戏引擎
行走__Wz5 小时前
计算机学习路线与编程语言选择(信息差)
java·开发语言·javascript·学习·编程语言选择·计算机学习路线
TUTO_TUTO5 小时前
【AWS+Wordpress】将本地 WordPress 网站部署到AWS
笔记·学习·云计算·aws
大溪地C6 小时前
CSS详细学习笔记
css·笔记·学习
LVerrrr7 小时前
Missashe考研日记-day32
学习·考研