Try installing triton only if linux & x86_64 (#1051)

openai · Mar 7, 2023 · 924e1f8 · 924e1f8
1 parent 4b0d5e5
commit 924e1f8
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -1,4 +1,5 @@
 import os
+import platform
 import sys
 
 import pkg_resources
@@ -11,8 +12,8 @@ def read_version(fname="whisper/version.py"):
 
 
 requirements = []
-if sys.platform.startswith("linux"):
-    triton_requirement = "triton>=2.0.0.dev20221202"
+if sys.platform.startswith("linux") and platform.machine() == "x86_64":
+    triton_requirement = "triton==2.0.0"
     try:
         import re
         import subprocess