YOLOv5安装（cuda11.8）

本文档测试电脑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 ，测试成功，去相应的文件夹即可找到结果