diff --git a/nvidia_requirements.txt b/nvidia_requirements.txt new file mode 100644 index 0000000..6a78172 --- /dev/null +++ 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 deleted file mode 100644 index 6786911..0000000 --- a/requirements.txt +++ /dev/null @@ -1,13 +0,0 @@ -# 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" - -# For AMD GPUs with DirectML (Windows) -torch==2.4.1; platform_system == "Windows" and platform_machine == "AMD64" and extra == "dml" -torchvision==0.19.1; platform_system == "Windows" and platform_machine == "AMD64" and extra == "dml" -torchaudio==2.4.1; platform_system == "Windows" and platform_machine == "AMD64" and extra == "dml" -onnx==1.17.0; platform_system == "Windows" and platform_machine == "AMD64" -onnxruntime-directml==1.19.2; platform_system == "Windows" and platform_machine == "AMD64"