WindowsのAnacondaでTritonがインストールできないのを解消する方法

AI環境構築に欠かせない「Triton」がrequirements.txtに入っているのにもかかわらず、インストールできないでお困りではありませんか？

実際私も「Champ」をインストールする際にrequirements.txtに「triton」が入っているにも関わらず、インストールができないという問題に直面していました。

今回はこれを解消する方法をお伝えします。

1. エラー内容

pip install triton==2.0.0

というのがrequirements.txtに入っていました。しかし、以下のようなエラーが出てしまいます。

ERROR: Could not find a version that satisfies the requirement triton==2.0.0 (from versions: none)
ERROR: No matching distribution found for triton==2.0.0

エラーを読む限り2.0.0バージョンが見つからないと記載されています。

では、他のバージョンでもいいからインストールできないか試みてみました。

pip install triton

しかし、

ERROR: Could not find a version that satisfies the requirement triton (from versions: none)
ERROR: No matching distribution found for triton

インストールできません。

2. 原因

https://pypi.org/project/triton

TritonのPyPlには確かに2.0.0バージョンはあります。しかしなぜインストールできないのでしょうか。

答えは、Tritonが、Linux用に展開されているパッケージだからです。

Windows版のTritonがないからインストールができないのです。

私もここで諦めかけました。しかし、一つのGitHubが目につきます。

3. 解決

https://github.com/PrashantSaikia/Triton-for-Windows

Windows用のTritonを作っている方がいました。

インストールのポイントを解説します。

まず、Pythonのバージョンは必ず3.10にすることです。

Anacondaでは

  conda create -n myvenv python=3.10
  conda activate myvenv

こうすることでmyvenvという仮想環境内でpython 3.10が適用されています。

次に、GitHubからレポジトリーをクローンします。

git clone https://github.com/PrashantSaikia/Triton-for-Windows.git

ディレクトリを変更します。

cd Triton-for-Windows

インストールします。

pip install triton-2.0.0-cp310-cp310-win_amd64.whl

これで完了です！

もし、triton 2.1.0をインストールしたい場合は

pip install https://huggingface.co/madbuda/triton-windows-builds/resolve/main/triton-2.1.0-cp310-cp310-win_amd64.whl

これで行うことができます。

以上です！