增加torch cuda版本

requirements_cpu.txt

@@ -0,0 +1,43 @@
+# aicheck_v2 requirements
+# Usage: pip install -r requirements_cpu.txt
+
+# API -------------------------------------------------------------------------
+fastapi~=0.74.1 # fastapi基础包
+pydantic~=1.9.0 # pydantic
+email-validator==1.1.3
+PyJWT~=2.3.0
+python-multipart==0.0.5
+redis~=4.1.4
+SQLAlchemy~=2.0.34
+uvicorn~=0.17.5
+uvicorn[standard]
+loguru~=0.6.0
+xlrd~=2.0.1
+bcrypt==3.2.0
+pymysql==1.0.2
+pynvml==12.0.0
+requests-toolbelt==1.0.0
+python-socketio == 5.12.1
+
+
+# YOLOV5 ----------------------------------------------------------------------
+# BASE ------------------------------------------------------------------------
+gitpython>=3.1.30
+matplotlib>=3.3
+numpy==2.0.2
+opencv-python>=4.1.1
+pillow>=10.3.0
+psutil  # system resources
+PyYAML>=5.3.1
+requests>=2.32.2
+scipy==1.13.1
+thop>=0.1.1  # FLOPs computation
+torch==2.3.0+cpu --extra-index-url https://download.pytorch.org/whl/cpu
+torchvision==0.18.0+cpu --extra-index-url https://download.pytorch.org/whl/cpu
+torchaudio==2.3.0+cpu --extra-index-url https://download.pytorch.org/whl/cpu
+tqdm>=4.66.3
+ultralytics>=8.2.34  # https://ultralytics.com
+
+# Plotting --------------------------------------------------------------------
+pandas==2.2.3
+seaborn>=0.11.0

requirements_gpu.txt

@@ -32,8 +32,9 @@ PyYAML>=5.3.1
 requests>=2.32.2
 scipy==1.13.1
 thop>=0.1.1  # FLOPs computation
-torch>=1.8.0  # see https://pytorch.org/get-started/locally (recommended)
+torch --extra-index-url https://download.pytorch.org/whl/cu121
-torchvision>=0.9.0
+torchvision --extra-index-url https://download.pytorch.org/whl/cu121
+torchaudio --extra-index-url https://download.pytorch.org/whl/cu121
 tqdm>=4.66.3
 ultralytics>=8.2.34  # https://ultralytics.com