將whisper換成使用CUDA的方法 - 2023

2024/02/03閱讀時間約 1 分鐘

1. 在命令提示字元輸入nvidia-smi

 ∟確認自己電腦真的有NVIDIA顯卡,同時更重要的是看自己顯卡的CUDA Version版本

 ∟我顯卡的CUDA Version版本是12.1

2. 到PyTorch官網 https://pytorch.org/

3. 滾輪滾一下找到INSTALL PYTORCH

4. 確認哪個PyTorch Build有支援你的CUDA Version版本

 ∟舉例來說我顯卡的CUDA Version版本是12.1,那我在PyTorch Build - Preview (Nightly)中可以看到她有支援12.1的CUDA版本

5. 確認之後點Package – Pip

6. 複製Run this Command:的一長串

7. 按照影片教學解壓縮原本安裝的PyTorch版本

8. 按照影片教學清除快取

9. 貼上你剛剛在PyTorch官網中複製的那一長串

10. 等它安裝完

11. 輸入whisper --help檢查你device DEVICE是使用cuda的

12. 完成!你可以輸入指令來讓whisper幫你做語音辨識了


------------------------------------------------------------------------------------確認顯卡支援版本nvidia-smi


解除PyTorch安裝pip uninstall torch


清除快取pip cache purge


PyTorch官網→https://pytorch.org/get-started/locally/

---------------------------------------------------------------------------------

YT影片

1會員
16內容數
想做的事情太多,去做的事情太少。
留言0
查看全部
發表第一個留言支持創作者!
從 Google News 追蹤更多 vocus 的最新精選內容