Use triton==2.0.0 (#1053)

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",