如何在Windows系统上使用CUDA运行Mozilla TTS/Coqui TTS培训? [英] How can I run Mozilla TTS/Coqui TTS training with CUDA on a Windows system?

查看:19
本文介绍了如何在Windows系统上使用CUDA运行Mozilla TTS/Coqui TTS培训?的处理方法,对大家解决问题具有一定的参考价值,需要的朋友们下面随着小编来一起学习吧!

问题描述

我有一台配备Quadro P5000显卡的计算机,运行Windows 10。我想在此系统上培训TTS语音。我需要安装什么才能使其正常工作?

推荐答案

以下是安装/操作:

  1. Download并安装用于Windows的Python3.8(3.9+)。在安装过程中,请确保:
  • 选择为所有用户安装它。
  • 选择将Python添加到路径。
  1. Download并安装CUDA工具包10.1(11.0+)。
  2. Download&cuDNN v7.6.5(2019年11月5日),对于CUDA 10.1";(cuDNN v8+),将其解压缩,然后将cuda文件夹中的内容复制到C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.1中。
  3. Download最新的64位版本的eSak NG(无版本限制:-)。
  4. DownloadGit for Windows的最新64位版本(无版本限制:-)。
  5. 打开PowerShell提示符,找到要安装Coqui TTS的文件夹。
  6. 运行git clone https://github.com/coqui-ai/TTS.git
  7. 运行cd TTS
  8. 运行python -m venv .
  9. 运行.Scriptspip install -e .
  10. 运行以下命令(这与您从the PyTorch website获得的命令不同,因为a known issue):
.Scriptspip install torch==1.8.0+cu101 torchvision==0.9.0+cu101 torchaudio===0.8.0 -f https://download.pytorch.org/whl/torch_stable.html
  1. 将以下内容放入TTS文件夹中名为";test_cuda.py";的脚本中:
import torch
x = torch.rand(5, 3)
print(x)
print(torch.cuda.is_available())
  1. 通过.Scriptspython ./test_cuda.py运行脚本并确认输出如下所示(第一部分应该只有随机数,但最后一行必须为True;如果没有,则表示CUDA安装不正确):
tensor([[0.2141, 0.7808, 0.9298],
        [0.3107, 0.8569, 0.9562],
        [0.2878, 0.7515, 0.5547],
        [0.5007, 0.6904, 0.4136],
        [0.2443, 0.4158, 0.4245]])
True
  1. 将以下内容放入TTS文件夹中名为";Train.bat";的脚本中,然后为您的配置文件自定义该脚本:
set PYTHONIOENCODING=UTF-8
set PYTHONLEGACYWINDOWSSTDIO=UTF-8
set PHONEMIZER_ESPEAK_PATH=C:/Program Files/eSpeak NG/espeak-ng.exe

.Scriptspython.exe ./TTS/bin/train_tacotron.py --config_path "C:/path/to/your/config.json"
  1. 通过. rain.bat运行脚本。

如果您使用的型号与Taco tron不同,或者需要将其他参数传递到培训脚本中,请随时进一步自定义train.bat

如果您刚刚开始一般的TTS培训,请浏览How do I get started training a custom voice model with Mozilla TTS on Ubuntu 20.04?

这篇关于如何在Windows系统上使用CUDA运行Mozilla TTS/Coqui TTS培训?的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持IT屋!

查看全文
登录 关闭
扫码关注1秒登录
发送“验证码”获取 | 15天全站免登陆