Pytorch 2.0 Yolov5推理性能提升记录

测试环境

Nvidia A30
CUDA 11.7
Python 3.9
Pytorch 2.0
Ubuntu 20.04
Yolov5 v7.0

测试结果记录:

模型 size compile Speed(ms/image) 显存占用(GB)
yolov5m 640 × 13.4 1.7
yolov5m 640 11.3 1.6
yolov5m 640 × 16.2 2.2
yolov5m 640 14.6 2.1

你可能感兴趣的:(pytorch,YOLO,ubuntu)