windows本地搭建openai whisper并開啟NVIDIA GPU加速

需要的工具

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/
packageconda
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
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容