需要的工具
anaconda:python環(huán)境管理工具
chocolatey:windows包管理工具
anaconda安裝無腦下一步就好
chocolatey安裝看官網(wǎng)文檔
安裝過程
生成python環(huán)境
打開 Anaconda Powershell Prompt
conda create -n whisper python=3.9
conda activate whisper
pip install git+https://github.com/openai/whisper.git
whisper video.mp4 --model medium --language Chinese
可能會提示需要安裝 ffmpeg
打開 Powershell 管理員模式
choco install ffmpeg
開啟GPU加速
安裝 pytorch
官方安裝教程:https://pytorch.org/get-started/locally/
package 選 conda
CUDA 版本選最新的 我這里是 CUDA 11.8
選好后,復(fù)制生成的命令,放到 Anaconda Powershell Prompt 運行
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
安裝 cuda
choco install cuda --version=11.8.0.52206 # 注意和上一步選擇的版本要一致。版本號可以從choco官網(wǎng)軟件商店查到。
!??!cuda的版本和pytorch選擇的版本要一致。
?。?!cuda的版本和pytorch選擇的版本要一致。
?。?!cuda的版本和pytorch選擇的版本要一致。
檢測 pytorch 是否正常
python 運行
import torch
print(torch.cuda.is_available()) # true 則開啟成功
如果是false 檢查下 cuda 版本,是否顯卡過老,安裝了不適用的cuda版本
cuda 和 gpu對應(yīng)版本
運行whisper
whisper video.mp4 --model medium --language Chinese --device cuda