IT News

Torch is not able to use GPU

post tistory 2023. 10. 24.
반응형

 

Stable Diffusion을 실행하는 데 문제가 있습니까? "토치가 GPU를 사용할 수 없습니다" 오류 메시지를 해결해야 합니까? 다른 사용자에게 도움이 된이 문제를 해결하는 방법을 설명하는이 게시물을 읽으십시오. 바라건대 그것은 당신에게도 도움이 될 것입니다!

 

이 문제에 대한 많은 응답이 있었지만 여러 사용자가 아래 단계를 통해 문제를 해결했다고 말하고 있습니다. Stable Diffusion은 딥 러닝 알고리즘의 확산 모델로 구성되므로 이름과 수신하는 모든 데이터로 인해 오류가 발생할 수 있습니다. 하지만 걱정하지 마세요,

Torch is not able to use GPU 해결 방법은 다음과 같습니다

1.         Stablediffusion 폴더로 이동합니다.

2.         "VENV" 폴더를 삭제합니다.

3.         "webui-user.bat"을 실행합니다. 이렇게 하면 VENV 폴더가 다시 설치됩니다. (이 작업은 몇 분 정도 걸릴 수 있습니다.)

4.         초기 webui.sh를 실행합니다.

5.         pip list를 실행합니다.

6.         torch torchvision +rocm5.2로 설치되어 있는지 확인합니다.

7.         pip uninstall으로 해당 버전을 제거합니다.

8.         nvidia-smi를 사용하여 CUDA 버전을 확인합니다.

9.         지침에 따라 torch 버전을 설치합니다.

10.       pip3 list를 사용하여 빠른 검증을 수행합니다.

11.       상단 메뉴 중 런타임 > 런타임 유형 변경 > GPU 변경 후 재시작합니다. 만약 '백엔드를 할당하지 못했습니다...' 메세지가 나온다면 할당량 초과된 것입니다.

 

반응형

댓글

💲 추천 글