Skip to content

Commit

Permalink
Use triton==2.0.0 (openai#1053)
Browse files Browse the repository at this point in the history
  • Loading branch information
jongwook authored and ilanit1997 committed May 16, 2023
1 parent 77a0a3f commit 07c4b0b
Showing 1 changed file with 1 addition and 15 deletions.
16 changes: 1 addition & 15 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -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",
Expand Down

0 comments on commit 07c4b0b

Please sign in to comment.