AssertionError: Torch is not able to use GPU; add –skip-torch-cuda-test to COMMANDLINE_ARGS variable to disable this check

2025年8月10日

Stable Diffusion WebUI

Stable Diffusion WebUIを動かした時にこのようなエラーが出ました

add –skip-torch-cuda-test to COMMANDLINE_ARGS の場合は GPU を使用することができません

確認

pip list 

torch                        2.6.0
torchvision                  0.21.0

次のようになっていたら良いです

pip list 

torch                        2.6.0+cu124
torchvision                  0.21.0+cu124

次のコマンドで利用可能か確認

python -c "import torch; print(torch.randn(1).cuda())" 

gives RuntimeError: No CUDA GPUs are available

cuda

nvcc --version


nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2024 NVIDIA Corporation
Built on Thu_Mar_28_02:30:10_Pacific_Daylight_Time_2024
Cuda compilation tools, release 12.4, V12.4.131
Build cuda_12.4.r12.4/compiler.34097967_0

インストールされていなかったら次のサイトからダウンロードしてインストール

CUDA Toolkit Archive | NVIDIA Developer

バージョンにあったpytorchをインストール

Get Started

pip3 uninstall torch torchvision
pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu124

バッチファイル

@echo off
call %~dp0\scripts\env_for_icons.bat  %*
SET PATH=%PATH%;%WINPYDIRBASE%\PortableGit;%WINPYDIRBASE%\PortableGit\bin
If not exist %WINPYDIRBASE%\content mkdir  %WINPYDIRBASE%\content 

set APP_NAME=stable-diffusion-webui
set APP_DIR=%WINPYDIRBASE%\content\%APP_NAME%
echo %APP_DIR%
cd %APP_DIR%
if not defined VENV_DIR (set "VENV_DIR=%APP_DIR%\venv")
if EXIST %VENV_DIR% goto :activate_venv

python.exe -m venv "%VENV_DIR%" 
if %ERRORLEVEL% == 0 goto :activate_venv
echo Unable to create venv 
goto :skip_venv

:activate_venv
call "%VENV_DIR%\Scripts\activate"
::仮装環境でコマンドプロンプトをする時
::cmd.exe /k
::初回インストールやライブラリーを更新する時
::call webui-user.bat  
::オフラインで起動する時
::call webui.bat  

::python.exe webui.py

:skip_venv

xformers Windows で使う場合

https://github.com/woct0rdho/triton-windows/releases

https://download.pytorch.org/whl

pip install https://github.com/woct0rdho/triton-windows/releases/download/v3.2.0-windows.post10/triton-3.2.0-cp310-cp310-win_amd64.whl
pip install -U xformers --index-url https://download.pytorch.org/whl/cu124

バッチファイルを修正

set COMMANDLINE_ARGS=--xformers

バッチファイル修正

set COMMANDLINE_ARGS=--xformers