cupy

2025-07-26 12:29:32 -07:00
parent ccc68a3895
commit 725a781456
3 changed files with 165 additions and 21 deletions
--- a/requirements.txt
+++ b/requirements.txt
@@ -9,4 +9,7 @@ ultralytics>=8.0.0
 tqdm>=4.65.0
 psutil>=5.9.0
 ffmpeg-python>=0.2.0
-decord>=0.6.0
+decord>=0.6.0
+# GPU acceleration (optional but recommended for stereo validation speedup)
+# cupy-cuda11x>=12.0.0  # For CUDA 11.x
+# cupy-cuda12x>=12.0.0  # For CUDA 12.x - uncomment appropriate version