搭建语音训练模型
2023-03-01 本文已影响0人
Balance_1d14
课题:依据chatGPT模型搭建语音的训练模型
需求: 离线语音转文字的训练模型
新途径:docker以及chatGPT
目标:
1.能提供一款实现语音转文字的训练模型
2.能提供一款文字转语音的训练模型
进度:根据chatGPT的提示搭建训练模型,并且正在训练中
成果:
详细的搭建步骤如下:
# 先下载好训练模型文件
cd /mnt/model
curl -LO https://github.com/mozilla/DeepSpeech/releases/download/v0.9.3/deepspeech-0.9.3-models.pbmm
curl -LO https://github.com/mozilla/DeepSpeech/releases/download/v0.9.3/deepspeech-0.9.3-models.scorer
# 下载好docker源镜像
docker pull mozilla/deepspeech
docker run -it --rm -v $PWD:/mnt/model mozilla/deepspeech:latest bash
在部署实施的时候发现会出现以下错误
bash: nvidia-smi: not found
这是因为centos机器没有安装英伟达驱动导致的,根据chatGPT的提示,我们需要做如下操作:
sudo echo "blacklist nouveau" >> /etc/modprobe.d/blacklist.conf
sudo echo "options nouveau modeset=0" >> /etc/modprobe.d/blacklist.conf
sudo dracut --force
# Add the NVIDIA package repositories
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.repo | \
sudo tee /etc/yum.repos.d/nvidia-docker.repo
sudo yum install -y nvidia-container-toolkit
sudo systemctl restart docker
最后我们执行以下操作即可部署一款简易训练模型:
docker run -it --rm \
-v $PWD:/mnt/data \
mozilla/deepspeech:latest \
deepspeech --model /mnt/data/deepspeech-0.9.3-models.pbmm \
--scorer /mnt/data/deepspeech-0.9.3-models.scorer \
--audio /mnt/data/my_audio_file.wav