前往
大廳
主題

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

藤煤竹color | 2024-02-03 12:09:27 | 巴幣 0 | 人氣 143

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官網
-------------------------------------------------------------------------------------------------

創作回應

更多創作