fix: Update PyTorch versions for Python 3.12 compatibility and use PyTorch index for GPU builds

2026-02-16 20:03:17 +01:00
parent a5e7c73379
commit 72e7cdcee0
3 changed files with 6 additions and 4 deletions
--- a/requirements-gpu.txt
+++ b/requirements-gpu.txt
@@ -1,4 +1,5 @@
-torch==2.2.0+cu124
-torchvision==0.17.0+cu124
+--index-url https://download.pytorch.org/whl/cu124
+torch==2.5.1
+torchvision==0.20.1
 Real-ESRGAN==0.3.0
 basicsr==1.4.2