需要的工具
anaconda:python环境管理工具
chocolatey:windows包管理工具
anaconda安装无脑下一步就好
chocolatey安装看官网文档
安装过程
生成python环境
打开 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
选好后,复制生成的命令,放到 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官网软件商店查到。
!!!cuda的版本和pytorch选择的版本要一致。
!!!cuda的版本和pytorch选择的版本要一致。
!!!cuda的版本和pytorch选择的版本要一致。
检测 pytorch 是否正常
python 运行
import torch
print(torch.cuda.is_available()) # true 则开启成功
如果是false 检查下 cuda 版本,是否显卡过老,安装了不适用的cuda版本
cuda 和 gpu对应版本
运行whisper
whisper video.mp4 --model medium --language Chinese --device cuda