windows本地搭建openai whisper并开启NVIDIA GPU加速

需要的工具

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/
packageconda
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
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

友情链接更多精彩内容