소스 검색

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
myhloli 11 달 전
부모
커밋
28dae58839
1개의 변경된 파일2개의 추가작업 그리고 0개의 파일을 삭제
  1. 2 0
      requirements-docker.txt

+ 2 - 0
requirements-docker.txt

@@ -8,6 +8,8 @@ fast-langdetect==0.2.0
 scikit-learn>=1.0.2
 pdfminer.six==20231228
 unimernet==0.2.2
+torch>=2.2.2,<=2.3.1
+torchvision>=0.17.2,<=0.18.1
 matplotlib
 ultralytics>=8.3.48
 paddleocr==2.7.3