Update amd_requirements.txt

Signed-off-by: Bubbles The Dev <[email protected]>
KernFerm · Oct 2, 2024 · fb57aec · fb57aec
1 parent 66d4624
commit fb57aec
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/amd_requirements.txt b/amd_requirements.txt
@@ -1,6 +1,7 @@
 # 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"
+torch==2.4.1
+torchvision==0.19.1
+torchaudio==2.4.1
+onnx==1.17.0
+onnxruntime-directml==1.19.2
+numpy