一. PyTorch环境配置及安装

安装Anaconda3参考：https://blog.csdn.net/qq_36015370/article/details/79484455
其他部分参考： https://blog.csdn.net/zhouzhiyao960211/article/details/102873717.

注意：3060显卡选择pytorch的v1.80版本的CUDA 11.1

# CUDA 11.1
conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge

图为Anaconda3默认安装位置
在这里插入图片描述

二. 安装YOLOv5依赖

1、按照 “requirements.txt” 安装依赖

//第一步
cd "requirements.txt"文件路径 
#例如：cd C:\Users\LZY\Desktop\yolov5\yolov5
//第二步
conda activate pytorch
//第三步
pip install -r requirements.txt

在这里插入图片描述

2、安装结果

安装结果给出了pycharm的python配置环境的路径
在这里插入图片描述

三. pycharm打开YOLOv5

1、打开 “文件—设置—Python解释器” 界面，配置python

在这里插入图片描述

2、将"当前文件夹"设置为"源根"

链接: https://blog.csdn.net/qq_30622831/article/details/80978118
这一步是解决：pycharm导入本地py文件时，模块下方出现红色波浪线
在这里插入图片描述

四. 简单测试

在这里插入图片描述
注意：运行v5报以下错误，参考链接:
https://blog.csdn.net/RooKichenn/article/details/120866650.

如果你用的是v5的话,就去v6里面的models/common.py里面去找到这个SPPF的类,把它拷过来到你这个Tags5的models/common.py里面,这样你的代码就也有这个类了,还要在Tags5的models/common.py里引入一个warnings包就可以了