yolov5简单教程(自用)

配置环境

1.使用anaconda创建一个yolov5使用的虚拟环境

conda create -n env_name（环境名称） python=3.7(对应的python版本号）
# conda create -n yolov5 python=3.8                                          

2.激活虚拟环境

activate yolov5

3.手动安装pytorch的gpu版本

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

4.检查pytorch的gpu版本是否可用

python
import torch 
print(torch.__version__) # pytorch版本
print(torch.version.cuda) # cuda版本
print(torch.cuda.is_available()) # 查看cuda是否可用
# crtl+z 回车  退出python环境

5.安装其他必要的环境

在yolov5项目下找到requirements.txt，将有关torch的注释掉(因为之前已经下过gpu版本的了)

使用下面的指令,