Create amd_requirements.txt

Signed-off-by: Bubbles The Dev <[email protected]>
KernFerm · Oct 2, 2024 · 9d1539e · 9d1539e
1 parent 688abb9
commit 9d1539e
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/amd_requirements.txt b/amd_requirements.txt
@@ -0,0 +1,6 @@
+# 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"