Yolov5——pytorch环境搭建

Yolov5——pytorch环境搭建：

一． 准备工作：Cuda 、Anaconda安装、yolov5源码下载

1.cuda版本查看：

如果没有下载cuda的话，自行百度即可。

查看本机电脑cuda版本的方法：

方法一：

1. 在电脑桌面右击选中NVDIA控制面板
   

2. 打开界面后点击左下角系统信息
   

3. 点击组件后，在产品名称一栏即可查看CUDA版本
   

方法二：

win+R打开cmd，输入nvidia-smi

附上cuda安装教程：cuda安装教程

2.annacoda安装：

使用百度等引擎搜索anaconda进入官网之后下载即可（建议使用安装包，省得自己再去配置环境变量，麻烦，一步到位）按照网上的教程一直安装就好了。

附上一篇大佬的安装教程：Anaconda的详细安装步骤图文并茂
附上annaconda3-5.2版本的安装包
百度云链接：下载地址
提取码：gs16

创建pytorch环境：
安装annaconda完之后，我们需要新建pytorch环境，（annaconda安装完成之后会自带一个base环境，里面包括了一些常用的python第三方库类，但并不包括torch，因此我们需要建立一个带有torch的python编译环境，当然，你也可以选择用base环境来安装torch，不过安装时要注意base环境下的python版本需要与torch版本对应）

打开cmd执行代码：

# 创建了名叫pytorch，python版本为3.7的新环境
conda create -n pytorch python=3.7  

3.下载YOLOv5 github项目

点击前往yolov5GitHub源码地址

把压缩包解压到指定目录就行。

二． 安装pytorch及yolov5相关第三方库

1.pytorch安装：

1. 如果你的电脑已经安装了cuda，可以按照你自己的cuda版本下载pytorch对应版本的离线whl文件进行即可。
   推荐使用离线安装，在线太慢了。。。。。。
   torch相关下载地址：https://download.pytorch.org/whl/torch_stable.html
   打开网页后CTRL+F 快速搜索需要安装的torch版本
   比如：我的cuda版本是11，创建pycorch环境的python版本是3.7则搜索：
   ctrl+f 输入cu11然后找到对应的cp37（cp37即python3.7）选择合适版本点击下载即可
   

   2.打开cmd执行代码进行安装即可：

   # 激活名为pytorch的环境
   activate pytorch
   # 跳转到安装好的whl文件目录下
   cd C:\Users\17394\Downloads
   # pip安装下载好的whl文件
   pip install torch-1.7.1+cu110-cp37-cp37m-win_amd64.whl
   pip install torchvision-0.8.2+cu110-cp37-cp37m-win_amd64.whl
   

   3.安装yolov5所需的第三方库：
   yolov5项目下载解压后有一个requirements.txt文件

   继续执行以下代码安装即可：

   pip install requirements.txt
   

   （安装pycocotools库需要先安装Cython）

三、配置pycharm打开yolov5项目

1.打开yolov5项目修改python解释器

1. CTRL+ALT+S 打开设置

2. 执行以下步骤
   
   
   
   
   如果找不到的话可以打开cmd执行以下代码：

   # 查找已经创建的环境
   conda info -e
   

到此yolov5的环境就已经搭建好了！！！
如果有写的不好的地方还请斧正！！！！