WSL-Ubuntu20.04训练环境配置

news/2024/8/22 6:59:37 标签: yolov8, ubuntu

1.YOLOv8训练环境配置

        训练环境配置的话就仍然以YOLOv8为例,来说明如何配置深度学习训练环境。这部分内容比较简单,主要是安装miniAnaconda以及安装torch和torchvision.

        首先是miniAnaconda的安装(参考官网的教程Miniconda — Anaconda ),执行以下指令即可:

mkdir -p ~/miniconda3
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O ~/miniconda3/miniconda.sh
bash ~/miniconda3/miniconda.sh -b -u -p ~/miniconda3
rm -rf ~/miniconda3/miniconda.sh

        然后初始化一下,重新打开后就直接是anaconda环境了.

~/miniconda3/bin/conda init bash

         最后利用anaconda创建YOLOv8训练环境,这里使用的环境是python-3.8,torch-1.13,torchvision-0.14,cuda版本也和WSL的cuda版本保持一致,选择11.6。下载链接:download.pytorch.org/whl/torch_stable.html,(注意:原本我这里安装版本是torch-1.12,torchvision-0.13,但是在验证的时候会报错,torch版本提升至1.13后又没有报错了,所以这里我就选择torch-1.13和torchvison-0.14)

         将下载好的torch和torchvision的whl文件拷贝到/home/download路径下(这里的图片版本是torch-1.12,torchvision-0.13,我没有更换,大家替换成自己对应版本就行).

        然后执行以下指令来完成v8环境配置。 

conda create -n v8_env python=3.8 -y
conda activate v8_env
pip install /home/download/torch-1.13.0+cu116-cp38-cp38-linux_x86_64.whl
pip install /home/download/torchvision-0.14.0+cu116-cp38-cp38-linux_x86_64.whl
pip install ultralytics

 2.YOLOv8训练环境验证

        先来看一下torch是否安装好,以及是否可以正常使用gpu,如果返回true和能调用gpu,说明torch安装好了,如下所示。

        接下来使用v8提供的预训练权重来预测下结果,看看v8的环境是否安装好。         先在链接(GitHub - ultralytics/ultralytics: NEW - YOLOv8 )中把YOLOv8源代码和目标检测的预训练权重下载下来,放到项目代码中,然后再使用下面指令就可以查看预测结果,同时判断自己的YOLOv8环境是否已经安装成功,其中path/to/best.pt替换成自己的路径。

        把ultralytics-main.zip这个压缩包放在/home路径下(放在哪里都行),解压缩ultralytics包.

        执行以下命令,其中bus.jpg,则可以打开链接bus.jpg自行下载,然后放在项目文件中,执行结果保存在runs/detect/predict中,同样可以直接在windows的wsl路径(\wsl.localhost\Ubuntu-2004\home\ultralytics-main\runs\detect\predict)下打开这个文件查看运行结果。 

yolo detect predict model=/home/ultralytics-main/yolov8n.pt source=/home/ultralytics-main/bus.jpg

3.YOLOv8训练自己的数据集 

        这一部分主要是使用vs-code远程连接WSL-ubuntu20.04,其他都可以直接参考在Windows环境下标注、训练、验证、预测及导出的步骤,可以直接参考这两篇公开的博客。

        LabelImg下载及目标检测数据标注

        YOLOv8的训练、验证、预测及导出[目标检测实践篇]

        此外,我在训练-验证的时候还遇到以下错误,nvrtc: error: invalid value for --gpu-architecture (-arch),参考了iassc gym nvrtc: error: invalid value for --gpu-architecture-CSDN博客,并没有解决问题,然后我更新torch版本为1.13,torchvison版本为0.14后,就可以正常训练了,理论上来说应该不会这样的,知道的朋友可以告诉一下。

 

 


http://www.niftyadmin.cn/n/5556246.html

相关文章

第2部分:物联网模式在行动

第2部分:物联网模式在行动 上一部分中详细介绍的模式可以混合和匹配,以实现物联网用例。本部分提供了这些模式的有效性证明,以满足不同领域的独特需求并实施用例,包括消费品和家庭自动化、零售、运输、制造和农业。 本部分包括以…

实验一:图像信号的数字化

目录 一、实验目的 二、实验原理 三、实验内容 四、源程序及结果 源程序(python): 结果: 五、结果分析 一、实验目的 通过本实验了解图像的数字化过程,了解数字图像的数据矩阵表示法。掌握取样(象素个…

[ACM独立出版] 2024年虚拟现实、图像和信号处理国际学术会议(VRISP 2024,8月2日-4)

2024年虚拟现实、图像和信号处理国际学术会议(VRISP 2024)将于2024年8月2-4日在中国厦门召开。 VRISP 2024将围绕“虚拟现实、图像和信号处理”的最新研究领域,为来自国内外高等院校、科学研究所、企事业单位的专家、教授、学者、工程师等提供…

k8s学习笔记——dashboard安装

重装了k8s集群后,重新安装k8s的仪表板,发现与以前安装不一样的地方。主要是镜像下载的问题,由于网络安全以及国外网站封锁的原因,现在很多镜像按照官方提供的仓库地址都下拉不下来,导致安装失败。我查了好几天&#xf…

【持续集成_05课_Linux部署SonarQube及结合开发项目部署】

一、Linux下安装SonarQube 1、安装sonarQube 前置条件:sonarQube不能使用root账号进行启动,所以需要创建普通用户及 其用户组 1)创建组 2)添加用户、组名、密码 3)CMD上传qube文件-不能传到home路径下哦 4&#xff09…

LVS+Nginx高可用集群---keepalived原理与实战

1.高可用集群架构keepalived双机主备原理 高可用:(HA) 部署nginx存在两台nginx。当主节点的nginx宕机停止服务的时候,nginx备用机起到跟nginx(主) keepalived的概念:解决单点故障;组件免费;可以实现高可用HA机制&…

macos上latex环境搭建(homebrew安装+vscode配置+ MacTex)和B站视频、网站、教程等相关资料推荐(Overleaf、公式预览网站)

安装及配置 本机环境 本人为macos,已经安装了homebrew和vscode。希望得到的效果是在vscode中编辑并预览latex文件 MacTex安装 首先,使用brew安装MacTex(新版本的brew已经将install和install --cask合并了) brew install mactex安装后一般会置于如下…

非法闯入智能监测摄像机:安全守护的新利器

在当今社会,安全问题愈发受到重视。随着科技的进步,非法闯入智能监测摄像机应运而生,成为保护家庭和财产安全的重要工具。这种摄像机不仅具备监控功能,还集成了智能识别和报警系统,能够在第一时间内检测到潜在的入侵行…