一、安装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文件
检测结果：