From 688abb98e4e68b257b5d2e7d0f38a36d3cb54c71 Mon Sep 17 00:00:00 2001 From: Bubbles The Dev <152947339+KernFerm@users.noreply.github.com> Date: Wed, 2 Oct 2024 03:59:57 -0400 Subject: [PATCH] Update and rename requirements.txt to nvidia_requirements.txt Signed-off-by: Bubbles The Dev <152947339+KernFerm@users.noreply.github.com> --- nvidia_requirements.txt | 6 ++++++ requirements.txt | 13 ------------- 2 files changed, 6 insertions(+), 13 deletions(-) create mode 100644 nvidia_requirements.txt delete mode 100644 requirements.txt 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"