Skip to content

Commit 38e990d

Browse files
authored
Use triton==2.0.0 (#1053)
1 parent 924e1f8 commit 38e990d

File tree

1 file changed

+1
-15
lines changed

1 file changed

+1
-15
lines changed

setup.py

Lines changed: 1 addition & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -13,21 +13,7 @@ def read_version(fname="whisper/version.py"):
1313

1414
requirements = []
1515
if sys.platform.startswith("linux") and platform.machine() == "x86_64":
16-
triton_requirement = "triton==2.0.0"
17-
try:
18-
import re
19-
import subprocess
20-
21-
version_line = (
22-
subprocess.check_output(["nvcc", "--version"]).strip().split(b"\n")[-1]
23-
)
24-
major, minor = re.findall(rb"([\d]+)\.([\d]+)", version_line)[0]
25-
if (int(major), int(minor)) < (11, 4):
26-
# the last version supporting CUDA < 11.4
27-
triton_requirement = "triton==2.0.0.dev20221011"
28-
except (IndexError, OSError, subprocess.SubprocessError):
29-
pass
30-
requirements.append(triton_requirement)
16+
requirements.append("triton==2.0.0")
3117

3218
setup(
3319
name="openai-whisper",

0 commit comments

Comments
 (0)