Stability MatrixのComfyUIにROCm 7.1.1を入れてみると？

Windows上でAMD搭載のLLMをPyTorchでデプロイする初心者向けガイド - AMD GPUOpen

の記事にWindowsでPytorchを導入する方法があったのでStability MatrixのComfyUIに導入したらチョットは安定するんかなと思ったが結果そうでもなさそう。

なので読まなくて良いが、既存のパッケージの一部をオーバーホールしてみたかったので備忘録として残す

📋 前提環境と準備

このバージョンはいくつか準備がいる

1. 専用ドライバーのインストール

通常のAdrenalin Editionではなく、PyTorch専用のドライバーが必要。 以下のAMD公式ページからダウンロード

これが入っていないと始まりません。既存のドライバーがある場合はクリーンインストール推奨。

2. Python 3.12 の用意

ROCm 7.1.1 は Python 3.12 を要求します。 Stability Matrixは最近になって、Pythonバージョンを変更できるようになったのでこれを使えばよし。 自分は環境をアクティブにしなくても、そこのPythonが表示されるんだろうと思っていたため、環境アクティブにせずPythonバージョンを確認し、何故か食い違うPythonバージョンに結構悩んだが、環境アクティブにしてバージョン見ればStability Matrixが入れたPythonと同じになるはずだ。

🛠️ インストール手順

Stability Matrixが導入したComfyUIのパッケージのフォルダ内で cmd を開き、仮想環境を有効化

venv\Scripts\activate

手順1: 既存のPyTorchを削除

既存のPyTorchなどは、一度きれいに削除します。

pip uninstall -y torch torchaudio torchvision

手順2: ROCm SDKのインストール

AMDのリポジトリから、ROCmのSDKコンポーネントをインストールします。 依存関係のエラー（ERROR: pip's dependency resolver...）が出ることがあります、インストール自体が Successfully installed となっていれば無視。

pip install --no-cache-dir https://repo.radeon.com/rocm/windows/rocm-rel-7.1.1/rocm_sdk_core-0.1.dev0-py3-none-win_amd64.whl
pip install --no-cache-dir https://repo.radeon.com/rocm/windows/rocm-rel-7.1.1/rocm_sdk_devel-0.1.dev0-py3-none-win_amd64.whl
pip install --no-cache-dir https://repo.radeon.com/rocm/windows/rocm-rel-7.1.1/rocm_sdk_libraries_custom-0.1.dev0-py3-none-win_amd64.whl
pip install --no-cache-dir https://repo.radeon.com/rocm/windows/rocm-rel-7.1.1/rocm-0.1.dev0.tar.gz

手順3: ROCm対応 PyTorch のインストール

以下のURL指定でインストールしてください。

pip install --no-cache-dir https://repo.radeon.com/rocm/windows/rocm-rel-7.1.1/torch-2.9.0+rocmsdk20251116-cp312-cp312-win_amd64.whl
pip install --no-cache-dir https://repo.radeon.com/rocm/windows/rocm-rel-7.1.1/torchaudio-2.9.0+rocmsdk20251116-cp312-cp312-win_amd64.whl
pip install --no-cache-dir https://repo.radeon.com/rocm/windows/rocm-rel-7.1.1/torchvision-0.24.0+rocmsdk20251116-cp312-cp312-win_amd64.whl

✅ 動作確認

Pythonを対話モードで起動。

python

以下のコードを入力

import torch
print(f"Torch Version: {torch.__version__}")
print(f"Is HIP available: {torch.cuda.is_available()}")
print(f"Device Name: {torch.cuda.get_device_name(0) if torch.cuda.is_available() else 'N/A'}")

成功時の出力例: * Torch Version: 2.9.0+rocmsdk20251116 * Is HIP available: True * Device Name: AMD Radeon RX 9060 XT (またはGPU名)

Is HIP available: True となれば完了

PIP LIST

こうするとrocmSDK周りは0.1.dev0という特殊な番号になる

パッケージ名 バージョン
rocm 0.1.dev0
rocm-sdk-core 0.1.dev0
rocm-sdk-devel 0.1.dev0
rocm-sdk-libraries-custom 0.1.dev0
rocm-sdk-libraries-gfx120X-all 7.11.0a20251129
safetensors 0.7.0
scipy 1.16.3
sentencepiece 0.2.1
setuptools 70.2.0
six 1.17.0
spandrel 0.4.1
SQLAlchemy 2.0.44
sympy 1.14.0
tokenizers 0.22.1
torch 2.9.0+rocmsdk20251116
torchaudio 2.9.0+rocmsdk20251116
torchsde 0.2.6
torchvision 0.24.0+rocmsdk20251116

🚀 ComfyUI の起動

Stability Matrixから起動し、コンソールログを確認します。

AMD arch: gfx1200
ROCm version: (7, 1)
Device: cuda:0 AMD Radeon RX 9060 XT : native

このようにGPUが正しく認識されていれば成功です。 なおStability Matrixが入れてくれるCumfyUIはROCm version: (7, 2)であった事から、結果的にはわざわざ古いバージョンを入れたことに気づいたがこれも勉強なので。 一応このCumfyUIは画像を生成できたが、起動するたびに毎回「K-Sampler」をコンパイル（ビルド）してるなど、挙動が狂動になってしまった。