Update and rename requirements.txt to nvidia_requirements.txt

Signed-off-by: Bubbles The Dev <[email protected]>
KernFerm · Oct 2, 2024 · 688abb9 · 688abb9
1 parent 7bce947
commit 688abb9
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 13 deletions.
diff --git a/nvidia_requirements.txt b/nvidia_requirements.txt
@@ -0,0 +1,6 @@
+# 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"
diff --git a/requirements.txt b/requirements.txt