Commit 28dae588 authored by myhloli's avatar myhloli
Browse files

build(docker): add torch and torchvision dependencies

- Add torch>=2.2.2,<=2.3.1 to requirements-docker.txt- Add torchvision>=0.17.2,<=0.18.1 to requirements-docker.txt
parent ef78819a
...@@ -8,6 +8,8 @@ fast-langdetect==0.2.0 ...@@ -8,6 +8,8 @@ fast-langdetect==0.2.0
scikit-learn>=1.0.2 scikit-learn>=1.0.2
pdfminer.six==20231228 pdfminer.six==20231228
unimernet==0.2.2 unimernet==0.2.2
torch>=2.2.2,<=2.3.1
torchvision>=0.17.2,<=0.18.1
matplotlib matplotlib
ultralytics>=8.3.48 ultralytics>=8.3.48
paddleocr==2.7.3 paddleocr==2.7.3
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment