Update nvidia_requirements.txt

Signed-off-by: Bubbles The Dev <[email protected]>
KernFerm · Oct 18, 2024 · b42f60e · b42f60e
1 parent 155f553
commit b42f60e
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/nvidia_requirements.txt b/nvidia_requirements.txt
@@ -1,7 +1,7 @@
 # For NVIDIA GPUs with CUDA support (Windows)
-torch==2.5.0
-torchvision==0.19.1+cu118
-torchaudio==2.5.0
+torch==2.5.0+cu118
+torchvision==0.20.1+cu118
+torchaudio==2.5.0+cu118
 onnx==1.17.0
 onnxruntime-gpu==1.19.2
 numpy  # Required for PyTorch and ONNX operations