whisper 설치시 gpu가속이 안되는 경우 (wsl2)

whisper --language Korean --device cuda [file] 을 하면되는데 cpu가속이 되는경우가 있다.

확인 방법은

 

python3 -c "import torch; print(torch.cuda.is_available())"

False인 경우 버젼을 확인해본다.

python3 -c "import torch; print(torch.__version__)"

 

https://developer.nvidia.com/cuda-11-6-0-download-archive

CUDA Toolkit 11.6 Downloads

를 통해서 cuda를 설치해야되는데, 오늘기준으로 보면 cuda 11.6을 설치해야되는걸로 보인다.

 

https://pytorch.org/get-started/previous-versions/

Previous PyTorch Versions

에서 확인해보면 

# CUDA 11.6
pip install torch==1.13.1+cu116 torchvision==0.14.1+cu116 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu116

라고 되어있으니 그대로 실행한뒤 (pip3을 사용했음) torch버젼을 확인해보면 제대로 설치되어있다.

kalstein@GAMEPC:~/aa_lecture/20241108$ python3 -c "import torch; print(torch.__version__)"
1.13.1+cu116

 

이후에 whisper를 실행해보면 월등히 빠른속도로 나온다.