Ver código fonte

Use triton==2.0.0 (#1053)

Jong Wook Kim 2 anos atrás
pai
commit
38e990d853
1 arquivos alterados com 1 adições e 15 exclusões
  1. 1 15
      setup.py

+ 1 - 15
setup.py

@@ -13,21 +13,7 @@ def read_version(fname="whisper/version.py"):
 
 requirements = []
 if sys.platform.startswith("linux") and platform.machine() == "x86_64":
-    triton_requirement = "triton==2.0.0"
-    try:
-        import re
-        import subprocess
-
-        version_line = (
-            subprocess.check_output(["nvcc", "--version"]).strip().split(b"\n")[-1]
-        )
-        major, minor = re.findall(rb"([\d]+)\.([\d]+)", version_line)[0]
-        if (int(major), int(minor)) < (11, 4):
-            # the last version supporting CUDA < 11.4
-            triton_requirement = "triton==2.0.0.dev20221011"
-    except (IndexError, OSError, subprocess.SubprocessError):
-        pass
-    requirements.append(triton_requirement)
+    requirements.append("triton==2.0.0")
 
 setup(
     name="openai-whisper",