windows本地搭建openai whisper并开启NVID

2023-05-05  本文已影响0人  写给我的公主的简书

需要的工具

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
上一篇下一篇

猜你喜欢

热点阅读