본문 바로가기
프로그램 에러

윈도우 스테이블디퓨전(forge) RuntimeError: Torch is not able to use GPU

by 젤리아드 2024. 9. 28.
반응형

윈도우에서 스테이블디퓨전(forge)을 실행했는데 다음 에러가 발생했습니다.

Traceback (most recent call last):
File "webui\launch.py", line 51, in <module> main()
File "webui\launch.py", line 39, in main prepare_environment()
File "webui\modules\launch_utils.py", line 431, in prepare_environment
raise RuntimeError(
RuntimeError: Torch is not able to use GPU; add --skip-torch-cuda-test to COMMANDLINE_ARGS variable to disable this check

RuntimeError: Torch is not able to use GPU; add --skip-torch-cuda-test to COMMANDLINE_ARGS variable to disable this check
forge 에러메시지

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

 

이 에러의 원인은 다양합니다.

NVIDIA 그래픽 카드가 아닌 그래픽 카드를 사용하고 있거나 NVIDIA 그래픽 카드이지만 오래된 모델의 그래픽 카드 또는 모델에는 문제가 없지만 그래픽 카드 드라이버 버전이 옛날 버전일 경우 이 에러가 발생할 수 있습니다.

이 중에서 가장 빠르게 체크할 수 있는 부분이 그래픽 카드 드라이버가 최신 버전인지 확인하는 것입니다. 

포털 사이트에 "NVIDIA 그래픽 카드 드라이버"라고 검색한 후 NVIDIA 공식 사이트에 접속합니다.

NVIDIA
드라이버
수동 드라이버 검색
제품, 제품 유형 또는 시리즈로 검색
제품 범주 선택
제품 시리즈 선택
제품 선택
운영 체제 선택
NVIDIA 그래픽 카드 드라이버

사용 중인 모델을 선택하면 최신 드라이버의 버전을 알 수 있습니다.

 

실행
프로그램, 폴더, 문서, 또는 인터넷 주소를 입력하여 해당 항목을 열 수 있습니다.
열기(O): cmd
cmd 실행

윈도우키+R 누른 후 cmd 명령어를 실행합니다.

 

C:\Windows\system32\cmd

C:\> nvidia-smi
+-----------------------------
❘ NVIDIA-SMI 456.71 Driver Version: 456.71
그래픽 카드 드라이버 버전 확인

"nvidia-smi" 명령어를 실행하면 현재 시스템에 설치된 그래픽 카드 드라이버의 버전이 출력됩니다.

현재 설치된 버전과 NVIDIA 공식 사이트에 표기된 최신 버전을 비교하여 최신 버전이 아닐 경우 다운로드해서 설치합니다.

현재 최신 버전이 500 이상 버전이기 때문에 456 버전 정도면 굉장히 오래된 드라이버라고 할 수 있습니다.

이렇게 오래된 드라이버를 사용하면 GPU 에러가 발생합니다.

 

그래픽 카드 드라이버를 최신 버전으로 업데이트 완료 후 스테이블디퓨전(forge)을 다시 실행하여 GPU 에러가 발생하는지 확인합니다.

 

에러가 해결되지 않을 경우

GPU 에러가 해결되지 않을 경우 GPU 모드를 끄고 CPU 모드로 실행할 수는 있습니다.

다만 CPU 모드이기 때문에 이미지 생성이 GPU 모드만큼 빠르지는 않습니다.

메모장 webui-user.bat

@echo off

set PYTHON=
set GIT=
set VENV_DIR=
set COMMANDLINE_ARGS=--skip-torch-cuda-test
webui-user.bat 편집

스테이블디퓨전 forge일 경우 설치 경로\webui 폴더에 있는 webui-user.bat 파일을 메모장으로 엽니다.

COMMANDLINE_ARGS"--skip-torch-cuda-test"를 추가한 후 저장합니다.

이 옵션 적용 후 스테이블디퓨전을 실행하면 GPU 에러는 발생하지 않으나 스테이블디퓨전이 CPU 모드로 작동하게 됩니다.

반응형

댓글