美文网首页
windows本地搭建openai whisper并开启NVID

windows本地搭建openai whisper并开启NVID

作者: 写给我的公主的简书 | 来源:发表于2023-05-05 00:48 被阅读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
    

    相关文章

      网友评论

          本文标题:windows本地搭建openai whisper并开启NVID

          本文链接:https://www.haomeiwen.com/subject/qdyfsdtx.html