本文档测试电脑GPU:RTX 3060、RTX 3070

一、anaconda创建虚拟环境

conda create -n yolov5 python=3.8
conda activate yolov5

二、安装pytorch

pytorch官网
选择cuda11.8,复制代码运行即可

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

!!似乎用pip下会很慢很慢,遂改为conda(conda记得改为清华源)

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

三、下载yolov8源码

YOLOv5的GitHub首页
下载zip文件,把源码放在你喜欢的地方

四、安装依赖

cd到刚下载的源码的根目录,安装requiement.txt文件,加个清华源

pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple

五、测试环境

去github首页,下载预训练集 链接
把下好的文件放到源码根目录下
回到conda的终端,在yolov5的环境下cd到源码的根目录,执行代码

python detect.py

反馈 Results saved to runs\\detect\\exp2 ,测试成功,去相应的文件夹即可找到结果

Logo

北京人形旗下天工造物具身智能开源社区,聚焦具身天工与慧思开物两大平台

更多推荐