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를 설치해야되는데, 오늘기준으로 보면 cuda 11.6을 설치해야되는걸로 보인다.
https://pytorch.org/get-started/previous-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를 실행해보면 월등히 빠른속도로 나온다.