TX2开发板上conda环境运行实现yolov5

版本:jetpack4.6              ubuntu18.04

之前想在系统上直接实现yolov5结果发现安装依赖包的时候老出错

于是便运用了Archiconda创建环境运行,前前后后折腾了快半个月,决定记录一下

TX2上系统与普通ubuntu不同,为arm版,(平时安装一些东西也要注意,不然会报错)在上面不能安装anaconda,要安装archiconda,常用运行指令都是一样的。

一.  首先在官网下载Archiconda3-0.2.3-Linux-aarch64.sh:官网链接
       下载好后cd到下载的文件夹下并运行:

./Archiconda3-0.2.3-Linux-aarch64.sh

之后点yes,安装完毕。

二. 创建环境

conda基本指令:基本指令使用

我直接创建了一个名为pytorch的环境(代码中的pytorch可以换为自己命名,python版本也可换,若不带=3.6,则系统自动选择版本)

conda create -n pytorch python=3.6

然后激活环境

conda activate pytorch

便进入环境了

三.  环境中安装包(以下所属都在自己创建的环境中实行)

1. 安装cuda

原本不想安装太新的版本,怕后面不好配置

看别的教程直接conda install cudatoolkit=10.2就是报错,

Solving environment: failed

PackagesNotFoundError: The following packages are not available from current channels:

  - cudatoolkit=10.2至今不知道为什么,有知道的小伙伴可以告诉我

然后直接

conda install cudatoolkit

自动安装了11.4.2版本,应该是最新的,下载过程很慢,也算是安装成功了。

2. 安装opencv

直接

conda install opencv

自动安装了4.5.3版本,好像也是最新的,也直接安装成功了。

3. 安装pytorch和torchvision(注意:此文章中的torch和torchvision为CPU版,运行yolov5s检测图片100ms左右,需要GPU版请移步)

注意:一开始安装了torch1.8.0和torchvision0.8.0,运行yolov5的detect.py程序时提示torchvision版本需>=0.8.1,然后系统自动把我安装的torchvision的0.8.0删除了,又自动下载了torchvision的0.10.0版本,不得不感叹一下运行命令行的智能

注意:torch和torchvision以及python三者的版本对应关系

 TX2开发板上conda环境运行实现yolov5_第1张图片

 找到关系后到:pytorch和torchvision下载

                              我下载了:torch-1.9.0-cp36-cp36m-linux_aarch64.whl

                              和torchvision-0.10.0-cp36-cp36m-manylinux2014_aarch64.whl

cp后面的为你创建的环境中python的版本

aarch结尾的适配咱们的TX2开发板)

下载完成后将两个.whl文件移动到你创建的环境文件下再pip我的是archiconda3/envs/pytorch

cd到这个文件夹下

然后

pip install torch-1.9.0-cp36-cp36m-linux_aarch64.whl
pip install torchvision-0.10.0-cp36-cp36m-manylinux2014_aarch64.whl

便安装成功了。

其余的先不用安装,因为运行detect.py时会自动安装一些包,比如opencv-python。

之后将你训练好的yolov5的整个文件夹(假设文件夹名为yolov5_1)复制到你创建的环境文件夹下,我的是archiconda3/envs/pytorch

然后cd到这个目录下:cd archiconda3/envs/pytorch/yolov5_1

直接运行

python3 detect.py --source data/images --weights yolov5s.pt

运行应该会一直提示你缺什么包,然后你就直接按照提示,缺了哪个包直接pip install 包名

(有的时候直接带上软件包名字不行,想要保险操作就将提示的复制下来再搜索一下,网上都有解决方法) 

到不提示缺少软件包之后z再运行,就会有一个自动下载requirements:什么什么

然后就能显示出检测结果了。

你可能感兴趣的:(ubuntu,pytorch,opencv,nvidia)