怎樣使用AMD顯卡進行so-vits-svc4的模型訓(xùn)練及推理

一、使用前準備
ubuntu22或者ubuntu20系統(tǒng):https://ubuntu.com/download/desktop
A卡驅(qū)動:https://www.amd.com/zh-hans/support
Anaconda:https://repo.anaconda.com/archive/Anaconda3-2023.03-Linux-x86_64.sh
so-vits-svc:https://github.com/svc-develop-team/so-vits-svc/archive/refs/heads/4.0.zip
checkpoint_best_legacy_ 500.pt:https://ibm.box.com/s/z1wgl1stco8ffooyatzdwsqn2psd9lrr
G_0.pth/D_0.pth:https://huggingface.co/Himawari00/so-vits-svc4.0-pretrain-models/tree/main
二、Ubuntu系統(tǒng)安裝
如果需要使用A卡做vits-svc訓(xùn)練的話需要安裝linux系統(tǒng),推薦使用ubuntu。ubuntu系統(tǒng)安裝方法去問谷歌娘或者度娘,不過需要注意的是ubuntu系統(tǒng)最好安裝在單獨的硬盤內(nèi)以免干擾到正常系統(tǒng)的運行,如果你能熟練安裝雙系統(tǒng)則可以跳過。
三、顯卡驅(qū)動和rocm安裝
1、下載自己型號的顯卡驅(qū)動,比如我的型號顯卡驅(qū)動文件是amdgpu-install_5.4.50403-1_all.deb,打開命令(root用戶)行執(zhí)行以下命令
2、執(zhí)行完以上的命令我們就把顯卡驅(qū)動和rocm安裝完了,然后我們還需要配置一個環(huán)境
3、所有步驟執(zhí)行完之后執(zhí)行下以下命令,確認驅(qū)動及rocm安裝到位
四、Conda的安裝以及環(huán)境配置
Conda是一個開源的、跨平臺的包和環(huán)境管理工具,可以用于在同一個機器上安裝不同版本的軟件包及其依賴,并能夠在不同的環(huán)境之間切換。它是 Anaconda 發(fā)行版的一部分,但也可以單獨使用。
1、下載Conda并授權(quán)
2、給Anaconda文件授權(quán)
3、執(zhí)行Anaconda安裝程序
安裝過程中讓你答yes或no的地方通通答yes,然后安裝時會詢問安裝在哪個目錄,建議安裝在/opt/anaconda3目錄下。上面步驟安裝完成后執(zhí)行以下命令
4、接下來創(chuàng)建一個python環(huán)境
5、切換到我們剛創(chuàng)建的環(huán)境下后就需要安裝vits-svc所需要的pip環(huán)境了,我們首先下載so-vits-svc,然后解壓到任意文件夾,打開文件夾內(nèi)的requirements.txt文件,將scipy==1.7.3改成scipy==1.8.1,刪除torch和torchaudio的兩行,然后保存。
6、安裝所需pip環(huán)境
五、so-vits-svc的安裝以及使用
在命令行內(nèi)進入[四-5]所解壓的文件夾,然后進行以下操作(建議查看官方文檔:https://github.com/svc-develop-team/so-vits-svc/blob/4.0/README_zh_CN.md)
1、下載G_0.pth/D_0.pth,放到logs/44k/文件夾下
2、下載checkpoint_best_legacy_ 500.pt,放到hubert文件夾下
3、把要訓(xùn)練的素材放到dataset_raw文件夾下(訓(xùn)練人的聲音)
4、把受推理的素材放到raw文件夾下(模仿對象的素材)
5、根據(jù)實際情況調(diào)整configs/config.json文件
6、重采樣至44100Hz單聲道
7、自動劃分訓(xùn)練集、驗證集,以及自動生成配置文件
8、生成hubert與f0
9、開始訓(xùn)練
10、使用訓(xùn)練的模型進行推理
六、注意事項
1、A卡訓(xùn)練只能使用linux系統(tǒng)
2、強烈推薦使conda進行使用,會避免很多環(huán)境問題
3、之后需要訓(xùn)練時一定要先切換到創(chuàng)建好的conda環(huán)境,不然會報錯