Windows 11 で Hallo2 を使う時の環境をいろいろさぐってみた（WindowsでTritonを使う方法も）

はじめに

前回の続きです。
touch-sp.hatenablog.com
動作可能な環境をいろいろさぐってみました。

動作環境と環境構築

Python 3.10 + CUDA 12.4

Windows 11
CUDA 12.4
Python 3.10

pip install torch==2.4.1 torchvision==0.19.1 torchaudio==2.4.1 xformers --index-url https://download.pytorch.org/whl/cu124
pip install -r https://raw.githubusercontent.com/dai-ichiro/myEnvironments/refs/heads/main/Hallo2/requirements_win_cu124.txt

（オプション）こちらから「triton-3.0.0-cp310-cp310-win_amd64.whl」をダウンロードしてインストールするとTritonも使えました。

Python 3.11 + CUDA 12.4

Windows 11
CUDA 12.4
Python 3.11

pip install torch==2.4.1 torchvision==0.19.1 torchaudio==2.4.1 xformers --index-url https://download.pytorch.org/whl/cu124
pip install -r https://raw.githubusercontent.com/dai-ichiro/myEnvironments/refs/heads/main/Hallo2/requirements_win_cu124.txt

（オプション）こちらから「triton-3.0.0-cp311-cp311-win_amd64.whl」をダウンロードしてインストールするとTritonも使えました。

Python 3.12 + CUDA 12.4

Windows 11
CUDA 12.4
Python 3.12

pip install torch==2.4.1 torchvision==0.19.1 torchaudio==2.4.1 xformers --index-url https://download.pytorch.org/whl/cu124
pip install -r https://raw.githubusercontent.com/dai-ichiro/myEnvironments/refs/heads/main/Hallo2/requirements_win_cu124.txt

（オプション）こちらから「triton-3.0.0-cp312-cp312-win_amd64.whl」をダウンロードしてインストールするとTritonも使えました。