win10下yolov5的cpu和gpu环境搭建

一、安装pytorch

1、官网地址https://pytorch.org/

2、根据自己的需求选择

 

 说明：

• PyTorch Build 就是选择下载稳定版（Stable）还是预览版（Preview），这里一般选择稳定版（Stable）
• YourOS 不多说，选择自己对应的操作系统类型
• Package：下载方式，教程提供的是 Conda 方式，这里我们使用 Pip 方式
• Language：即选择你的 python 版本（Anaconda 下载后自带 python ，可以在 cmd 输入 python --version 查看自己的 python 版本）
• CUDA：如果要下载 gpu 版本，那么就选择你对应的 cuda 的版本（不知道自己 cuda 版本的请戳：https://www.jianshu.com/p/d3b9419a0f89）；如果下载 cpu 版本，那么选择 None
• Run this Command：当你选择好版本后，会自动再次生成下载安装 PyTorch 的命令

3、安照命令安装：

 

 二、下载YOLOv5 github项目

1、地址：https://github.com/ultralytics/yolov5

2、验证：

import torch
#查看版本
print(torch.__version__)
#查看gpu是否可用
print(torch.cuda.is_available())
#返回设备gpu个数
print(torch.cuda.device_count())

 

 3、运行detect.py文件

检测结果：