LOVOv5运行环境搭建

一、搭建运行环境

1.1 安装GPU环境

（1）下载和安装nvidia显卡驱动

安装之后，在cmd中输入：nvidia-smi，出现下图说明安装成功。

（2）下载CUDA

进入官网选择相应的版本下载

cuda下载链接：CUDA Toolkit Archive | NVIDIA Developer

（3）下载cuDNN

需要下载和CUDA对应的版本

cuDNN下载地址： CUDA Deep Neural Network (cuDNN) | NVIDIA Developer

（4）安装CUDA

配置环境变量

 （5）安装cuDNN

解压，将bin、include、lib三个文件复制到CUDA文件夹下。

（6）安装测试

打开cmd，输入：nvcc -V

 1.2 安装Anaconda

（1）下载安装包

Anaconda下载windows版：http://www.anaconda.com/products/individual

（2）安装Anaconda

（3）添加国内镜像

1.3 安装PyTorch

1.4 下载YOLOv5模型 

（1）下载地址：GitHub - ultralytics/yolov5: YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite

（2）安装所需库：

在YOLOv5路径下执行：pip install -r requirements.txt

（3）下载预训练权重文件