赞
踩
环境依然使用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文件夹中,如果没有该文件夹请自建一个)。
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。