【Yolov5 学习】

一，环境安装

1.安装 Anaconda （安装教程：b站）

2. Pytorch安装

https://pytorch.org/

conda creat -n yolov5 python=3.8
conda activate yolov5
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch-lts -c conda-forge

测试

import torch
torch.cuda.is_available()
torch.cuda.current_device()
torch.cuda.device(0)
torch.cuda.device_count()
torch.cuda.get_device_name(0)

3. Yolov5安装

https://github.com/ultralytics/yolov5

下载 源码 （Source code.zip）

在 requirements.txt 修改

numpy==1.20.3
Pillow==8.3.0
# torch>=1.7.0  # see https://pytorch.org/get-started/locally (recomme