Update requirements.txt

requirements.txt

@@ -1,7 +1,7 @@
---extra-index-url https://download.pytorch.org/whl/cu118
-torch
-torchvision==0.18.0
-torchaudio==2.3.0
+--extra-index-url https://download.pytorch.org/whl/cu128
+torch==2.11.0+cu128
+torchvision==0.22.0+cu128
+torchaudio==2.11.0+cu128
 
 numpy==1.26.4
 pyyaml==6.0.1
@@ -12,8 +12,8 @@ lmdb==1.4.1
 tqdm==4.66.4
 rich==13.7.1
 ffmpeg==1.4
-onnxruntime-gpu==1.18.0
-onnx==1.16.1
+onnxruntime-gpu==1.21.0
+onnx==1.17.0
 scikit-image==0.24.0
 albumentations==1.4.10
 matplotlib==3.9.0