GPU测试
This commit is contained in:
22
benchmark/requirements.txt
Normal file
22
benchmark/requirements.txt
Normal file
@@ -0,0 +1,22 @@
|
||||
# FP16 性能评估 Benchmark 框架依赖
|
||||
|
||||
# 核心依赖
|
||||
numpy>=1.21.0
|
||||
opencv-python>=4.5.0
|
||||
pyyaml>=6.0
|
||||
|
||||
# NVIDIA GPU 监控
|
||||
pynvml>=11.0.0
|
||||
|
||||
# Ultralytics YOLO
|
||||
ultralytics>=8.0.0
|
||||
|
||||
# TensorRT (需要单独安装)
|
||||
# tensorrt>=8.6.0
|
||||
|
||||
# PyCUDA (需要单独安装)
|
||||
# pycuda>=2022.1
|
||||
|
||||
# 可选: 数据分析
|
||||
# pandas>=1.3.0
|
||||
# matplotlib>=3.4.0
|
||||
Reference in New Issue
Block a user