yolo5环境搭建，Anaconda，pytorch GPU版本，cuda，cudnn安装教程

首先安装Anaconda----------------------------------------------

先将电脑里的python卸载，因为安装anaconda会自带一个基础版本的python，因此需要卸载，选择清华镜像源进行安装https://mirrors.bfsu.edu.cn/anaconda/archive/

选择Anaconda3-2020.07-Windows-x86_64.exe，双击安装包

选择所有用户

安装路径改成除C盘外的其他盘即可，因为后面要创建不同的环境，占用的空间还是很大的，所以建议不要安装大系统盘，接着

一路next，最后测试一下安装是否成功，win+R，输入cmd

在输入conda -V

安装完毕后需要更改一下镜像源

cmd输入:

conda config --add channels Index of /anaconda/pkgs/main/ | 北京外国语大学开源软件镜像站 | BFSU Open Source Mirror

此时C:\Users\xxxxx目录下会多一个.condarc的文件，文本打开改为下图即可

我们主要用的是Anaconda Prompt，打开

创建一个新的环境conda create --name yolo python-3.7

进入该环境，conda activate yolo

至此，Anaconda安装完成。

CUDA CUDnn安装------------------------------------------------------------------------------------------

确认一下支持的版本，win+R -->cmd-->nvidia-smi

CUDA-->https://developer.nvidia.com/cuda-toolkit-archive,我下的11.6

cuDnn->https://developer.nvidia.com/rdp/cudnn-download(相当于一个补丁，把它里面的文件复制到CUDA的目录下即可)

验证cudnn是否安装成功

cmd依次输入

->cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\extras\demo_suite

->bandwidthTest.exe

->deviceQuery.exe

pytorch GPU版本安装-----------------------------------------------------------

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

 安装完成后进行测试，依次输入

python

import torch

print(torch.__version__)

print(torch.version.cuda)

torch.cuda.is_available()    

print(torch.cuda.device_count())

若安装成功，则为下图所示：

卸载命令

pip uninstall torchaudio

pip uninstall torchvision

pip uninstall torch

Yolo下载-----------------------------------------------------------------

安装好git，新建一个文件夹，右键Git Bash here，输入：

git clone GitHub - ultralytics/yolov5: YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite

目录如下

python -m pip install -r requirements.txt，如果安装失败则需要不停的重试，我试了一下午才装好;

测试一下python detect.py --source ./data/images/bus.jpg

也可以测试视频python detect.py --source ./data/video/1.mp4

可以看到使用的是gpu版本

使用摄像头

Python detect.py --source 0（dog警告）

至此，yolo环境搭建完成；

踩坑记录------------------------------------------------------------

此类报错是因为版本不一致，解决办法如下：

-->pip install typing_extensions

目录:-->D:\Anaconda3\envs\pytorch\Lib\site-packages\torchvision\models\maxvit.py

将OrdereDict由从typing中输入改为typing_extensions即可