macsplex.com 로그인

검색

torch-is-not-able-to-use-gpu-error-v0-885k1xxhhenc1.png

Stable Diffusion webui.bat 실행시 torch is not able to use gpu가 나올시 다음과 같이 실행후 될수 있습니다.

Nvidia RTX A2000 드라이버버전 : 473.47 cuda : 11.4에서 테스트하였습니다.

 

1) torch 삭제

pip uninstall torch torchvision torchaudio
pip cache purge​*

 

 

2) cuda-python 및 Torch cuda 패키지를 설치한다. 

 

pip install cuda-python 
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

 

수정한 이유 :