当前位置:   article > 正文

YOLO系列:YOLOv8环境配置,安装,训练及测试流程_yolo v8训练最低配置

yolo v8训练最低配置

一、环境安装

环境依然使用anaconda配置虚拟环境,请确保虚拟环境的python>=3.7,torch>=1.7,我推荐python=3.8。同时,为了加速训练过程,有GPU平台的读者请确保CUDA>=10.2。

conda create -n 环境名字 python=3.8

我们假设你已经安装好了python=3.8的conda虚拟环境,进入刚刚建立好的虚拟环境后,安装GPU版本的torch和torchvision(没有GPU的请去掉+cu111)。

pip install torch==1.9.1+cu111 torchvision==0.10.1+cu111 -f https://download.pytorch.org/whl/torch_stable.html

二、安装

 下面进行下载YOLOv8,下面给出的是下载链接(可以顺道把预训练权重也下载下来,放在根目录下weights文件夹中,如果没有该文件夹请自建一个)。

https://www.wpsshop.cn/w/IT小白/article/detail/191566?site

推荐阅读
相关标签
  

闽ICP备14008679号