tighten requirements for rknn-toolkit2

pull/13506/head
Edward Yang 2025-02-03 10:46:14 +11:00
parent 3966484504
commit 89167fe774
1 changed files with 3 additions and 3 deletions

View File

@ -12,8 +12,8 @@ PyYAML>=5.3.1
requests>=2.23.0
scipy>=1.4.1
thop>=0.1.1 # FLOPs computation
torch>=1.7.0,<2 # see https://pytorch.org/get-started/locally (recommended)
torchvision>=0.8.1,<0.15
torch>=1.10.1,<2
torchvision>=0.11,<0.15
tqdm>=4.64.0
# protobuf<=3.20.1 # https://github.com/ultralytics/yolov5/issues/8012
@ -36,7 +36,7 @@ onnx-simplifier>=0.4.1 # ONNX simplifier
# tensorflow>=2.4.1 # TF exports (-cpu, -aarch64, -macos)
# tensorflowjs>=3.9.0 # TF.js export
# openvino-dev # OpenVINO export
rknn-toolkit2
rknn-toolkit2==2.3
# Deploy ----------------------------------------------------------------------
setuptools>=65.5.1 # Snyk vulnerability fix