yolov5环境搭建

Yolov5环境搭建

本机环境：

Python版本：3.9.13

Cuda版本：11.7

Torch版本：1.13.1

torchvision版本：0.14.1

Torch、torchvision、python版本要对应

1.下载yolov5项目GitHub - ultralytics/yolov5: YOLOv5 ��� in PyTorch > ONNX > CoreML > TFLite

2.安装Cuda

3.修改环境变量

用户变量---->Path---->添加五条：

(本机路径，按自己路径填入)

G:\xxxx\CUDA10.2\CUDA-Development\bin

G:\xxxx\CUDA10.2\CUDA-Development\libnvvp

G:\xxxx\CUDA10.2\CUDA-Development\extras\CUPTI\lib64

G:\xxxx\CUDA10.2\CUDA-Development\extras\CUPTI\include

G:\xxxx\CUDA10.2\CUDA-Development\lib\x64

4.下载cudnn:官网下载

下载完后把cudnn中的文件复制到cuda对应路径下

（bin、include、lib）

5.安装Anaconda:官网下载

Free Download | Anaconda

6.创建虚拟环境cmd中输入：conda create -n name python=3.9

查看anaconda中的虚拟环境：conda info -e

进入虚拟环境---->conda activate name

name为虚拟环境的名称  python=3.9是要在虚拟环境中使用的python版本

7.下载安装torch、torchvision

下载路径：download.pytorch.org/whl/torch_stable.html

cmd安装：pip install 路径\torch-1.9.1+cu102-cp39-cp39-win_amd64.whl

pip install 路径\torchvision-0.14.1+cu117-cp39-cp39-win_amd64.whl

Torch、torchvision、python版本要对应!!!!

8.安装好后cmd输入python -------> improt torch as t --------> t.__version__

安装成功

9.安装python

官网下载对应版本3.9

10.打开项目文件夹修改

与第7步下图对照

11.进入项目

12.无法导入的包需要自己导入

进入终端下载包：pip install xxxx

学完记得点个修赞^_^