久しぶりにStable DiffusionやWANで遊ぼうと思ったら、GPUがCUDAのサポート切れではまってしまったので、備忘録を作成しておきます。
対象
- PascalアーキテクチャのGPU
GeForce GTX 1080、GTX 1070、GTX 1060、GTX 1050 など - MaxwellアーキテクチャのGPU
GeForce GTX TITAN X、GTX 980、GTX 970、GTX 960 など
なぜはまるか
- これらGPUは、CUDA Toolkit 13.x以降でサポートされない
- これらGPUを搭載したマシンに、CUDA Toolkit 13.xや、対応のPyTorchをインストールできてしまうが、実行時に以下のようなエラーが発生する
torch.AcceleratorError: CUDA error: no kernel image is available for execution on the device
対処
- CUDA Toolkitを12.9.1に下げる
ここからダウンロード → https://developer.nvidia.com/cuda-12-9-1-download-archive - PyTorchを入れなおす
python -m pip uninstall torch torchvision torchaudio
python -m pip install torch==2.8.0+cu128 torchvision==0.23.0+cu128 torchaudio==2.8.0+cu128 --extra-index-url https://download.pytorch.org/whl/cu128
補足
- PyTorch 2.7.0 と書かれている場合が多いが、WANが2.2からデフォルトでdynamic VRAM有効になったので、対応できる2.8.0のほうが良い(たぶん)
- ComfyUIは自前でPythonを持っているので、そのフォルダ(デフォルトは C:\Users\XXXX\Documents\ComfyUI\.venv\Scripts)(XXXXはユーザ名)にcdして同様に処置する
- Stable Diffusion Web UIは、webui-user.batに以下の行を追加しておく
set TORCH_COMMAND=pip install -U torch==2.8.0+cu128 torchvision==0.23.0+cu128 torchaudio==2.8.0+cu128 --extra-index-url https://download.pytorch.org/whl/cu128
- NVIDIAドライバは最新で良い
参考