Update nvidia_requirements.txt

Signed-off-by: Bubbles The Dev <[email protected]>
KernFerm · Nov 1, 2024 · de82099 · de82099
1 parent 94af504
commit de82099
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nvidia_requirements.txt b/nvidia_requirements.txt
@@ -1,6 +1,6 @@
 # For NVIDIA GPUs with CUDA support (Windows)
 torch==2.5.1+cu118
-torchvision==0.20.1
+torchvision==0.20.1+cu118
 torchaudio==2.5.1+cu118
 onnx==1.17.0
 onnxruntime-gpu==1.19.2