Update nvidia_requirements.txt

Signed-off-by: Bubbles The Dev <152947339+KernFerm@users.noreply.github.com>
KernFerm · Oct 2, 2024 · 5300e86 · 5300e86
1 parent fb57aec
commit 5300e86
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/nvidia_requirements.txt b/nvidia_requirements.txt
@@ -1,6 +1,8 @@
 # For NVIDIA GPUs with CUDA support (Windows)
-torch==2.4.1+cu118; platform_system == "Windows" and platform_machine == "AMD64"
-torchvision==0.19.1+cu118; platform_system == "Windows" and platform_machine == "AMD64"
-torchaudio==2.4.1+cu118; platform_system == "Windows" and platform_machine == "AMD64"
-onnx==1.17.0; platform_system == "Windows" and platform_machine == "AMD64"
-onnxruntime-gpu==1.19.2; platform_system == "Windows" and platform_machine == "AMD64"
+torch==2.4.1+cu118
+torchvision==0.19.1+cu118
+torchaudio==2.4.1+cu118
+onnx==1.17.0
+onnxruntime-gpu==1.19.2
+numpy  # Required for PyTorch and ONNX operations
+scipy  # Optional, useful for scientific computing if needed