piper icon indicating copy to clipboard operation
piper copied to clipboard

Trouble Installing requirements on Ubuntu 24.04 for training

Open danielw97 opened this issue 7 months ago • 1 comments

Hi all, I'm interested in training a new piper voice, and am trying to get things working on ubuntu 24.04 (wsl 2) Python version is 3.10, and I've been following the instructions in#24 4 However, even though I've installed everything correctly (I believe) I'm receiving the following error when I try to initiate training: Has anyone seen this, and know how to solve it? Thanks in advance.

Traceback (most recent call last):                                                                                      
  File "/usr/lib/python3.10/runpy.py", line 196, in _run_module_as_main                                                 
    return _run_code(code, main_globals, None,                                                                          
  File "/usr/lib/python3.10/runpy.py", line 86, in _run_code                                                            
    exec(code, run_globals)                                                                                             
  File "/home/daniel/piper/src/python/piper_train/__main__.py", line 147, in <module>                                   
    main()                                                                                                              
  File "/home/daniel/piper/src/python/piper_train/__main__.py", line 124, in main                                       
    trainer.fit(model)                                                                                                  
  File "/home/daniel/piper/src/python/.venv/lib/python3.10/site-packages/pytorch_lightning/trainer/trainer.py", line 608, in fit                                                                                                                
    call._call_and_handle_interrupt(                                                                                    
  File "/home/daniel/piper/src/python/.venv/lib/python3.10/site-packages/pytorch_lightning/trainer/call.py", line 38, in _call_and_handle_interrupt                                                                                             
    return trainer_fn(*args, **kwargs)                                                                                  
  File "/home/daniel/piper/src/python/.venv/lib/python3.10/site-packages/pytorch_lightning/trainer/trainer.py", line 650, in _fit_impl                                                                                                          
    self._run(model, ckpt_path=self.ckpt_path)                                                                          
  File "/home/daniel/piper/src/python/.venv/lib/python3.10/site-packages/pytorch_lightning/trainer/trainer.py", line 1112, in _run                                                                                                              
    results = self._run_stage()                                                                                         
  File "/home/daniel/piper/src/python/.venv/lib/python3.10/site-packages/pytorch_lightning/trainer/trainer.py", line 1191, in _run_stage                                                                                                        
    self._run_train()                                                                                                   
  File "/home/daniel/piper/src/python/.venv/lib/python3.10/site-packages/pytorch_lightning/trainer/trainer.py", line 1204, in _run_train                                                                                                        
    self._run_sanity_check()                                                                                            
  File "/home/daniel/piper/src/python/.venv/lib/python3.10/site-packages/pytorch_lightning/trainer/trainer.py", line 1276, in _run_sanity_check                                                                                                 
    val_loop.run()                                                                                                      
  File "/home/daniel/piper/src/python/.venv/lib/python3.10/site-packages/pytorch_lightning/loops/loop.py", line 199, in run                                                                                                                     
    self.advance(*args, **kwargs)                                                                                       
  File "/home/daniel/piper/src/python/.venv/lib/python3.10/site-packages/pytorch_lightning/loops/dataloader/evaluation_loop.py", line 152, in advance                                                                                           
    dl_outputs = self.epoch_loop.run(self._data_fetcher, dl_max_batches, kwargs)                                        
  File "/home/daniel/piper/src/python/.venv/lib/python3.10/site-packages/pytorch_lightning/loops/loop.py", line 199, in run                                                                                                                     
    self.advance(*args, **kwargs)                                                                                       
  File "/home/daniel/piper/src/python/.venv/lib/python3.10/site-packages/pytorch_lightning/loops/epoch/evaluation_epoch_loop.py", line 137, in advance                                                                                          
    output = self._evaluation_step(**kwargs)                                                                            
  File "/home/daniel/piper/src/python/.venv/lib/python3.10/site-packages/pytorch_lightning/loops/epoch/evaluation_epoch_loop.py", line 234, in _evaluation_step                                                                                 
    output = self.trainer._call_strategy_hook(hook_name, *kwargs.values())                                              
  File "/home/daniel/piper/src/python/.venv/lib/python3.10/site-packages/pytorch_lightning/trainer/trainer.py", line 1494, in _call_strategy_hook                                                                                               
    output = fn(*args, **kwargs)                                                                                        
  File "/home/daniel/piper/src/python/.venv/lib/python3.10/site-packages/pytorch_lightning/strategies/strategy.py", line 390, in validation_step                                                                                                
    return self.model.validation_step(*args, **kwargs)                                                                  
  File "/home/daniel/piper/src/python/piper_train/vits/lightning.py", line 302, in validation_step                      
    self.logger.experiment.add_audio(                                                                                   
AttributeError: 'ExperimentWriter' object has no attribute 'add_audio'                                                  

In case it helps, here's my pip list as well:

aiohappyeyeballs         2.6.1                                                                                          
aiohttp                  3.11.14                                                                                        
aiosignal                1.3.2                                                                                          
async-timeout            5.0.1                                                                                          
attrs                    25.3.0                                                                                         
audioread                3.0.1                                                                                          
certifi                  2025.1.31                                                                                      
cffi                     1.17.1                                                                                         
charset-normalizer       3.4.1                                                                                          
cmake                    4.0.0                                                                                          
coloredlogs              15.0.1                                                                                         
Cython                   0.29.37                                                                                        
decorator                5.2.1                                                                                          
filelock                 3.18.0                                                                                         
flatbuffers              25.2.10                                                                                        
frozenlist               1.5.0                                                                                          
fsspec                   2025.3.1                                                                                       
humanfriendly            10.0                                                                                           
idna                     3.10                                                                                           
Jinja2                   3.1.6                                                                                          
joblib                   1.4.2                                                                                          
lazy_loader              0.4                                                                                            
librosa                  0.11.0                                                                                         
lightning-utilities      0.14.2                                                                                         
lit                      18.1.8                                                                                         
llvmlite                 0.44.0                                                                                         
MarkupSafe               3.0.2                                                                                          
mpmath                   1.3.0                                                                                          
msgpack                  1.1.0                                                                                          
multidict                6.2.0                                                                                          
networkx                 3.4.2                                                                                          
numba                    0.61.0                                                                                         
numpy                    1.24.4                                                                                         
nvidia-cublas-cu12       12.4.5.8                                                                                       
nvidia-cuda-cupti-cu12   12.4.127                                                                                       
nvidia-cuda-nvrtc-cu12   12.4.127                                                                                       
nvidia-cuda-runtime-cu12 12.4.127                                                                                       
nvidia-cudnn-cu12        9.1.0.70                                                                                       
nvidia-cufft-cu12        11.2.1.3                                                                                       
nvidia-curand-cu12       10.3.5.147                                                                                     
nvidia-cusolver-cu12     11.6.1.9                                                                                       
nvidia-cusparse-cu12     12.3.1.170                                                                                     
nvidia-cusparselt-cu12   0.6.2                                                                                          
nvidia-nccl-cu12         2.21.5                                                                                         
nvidia-nvjitlink-cu12    12.4.127                                                                                       
nvidia-nvtx-cu12         12.4.127                                                                                       
onnx                     1.17.0                                                                                         
onnxruntime              1.21.0                                                                                         
packaging                24.2                                                                                           
pip                      23.3.1                                                                                         
piper-phonemize          1.1.0                                                                                          
piper_train              1.0.0       /home/daniel/piper/src/python                                                      
platformdirs             4.3.7                                                                                          
pooch                    1.8.2                                                                                          
propcache                0.3.1                                                                                          
protobuf                 6.30.2                                                                                         
pycparser                2.22                                                                                           
pytorch-lightning        1.9.5                                                                                          
PyYAML                   6.0.2                                                                                          
requests                 2.32.3                                                                                         
scikit-learn             1.6.1                                                                                          
scipy                    1.15.2                                                                                         
setuptools               75.6.0                                                                                         
soundfile                0.13.1                                                                                         
soxr                     0.5.0.post1                                                                                    
sympy                    1.13.1                                                                                         
threadpoolctl            3.6.0                                                                                          
torch                    2.0.0+cu118                                                                                    
torchaudio               2.0.0+cu118                                                                                    
torchmetrics             0.11.4                                                                                         
tqdm                     4.67.1                                                                                         
triton                   2.0.0                                                                                          
typing_extensions        4.13.0                                                                                         
urllib3                  2.3.0                                                                                          
wheel                    0.45.1                                                                                         
yarl                     1.18.3                                                                                         

danielw97 avatar Mar 31 '25 15:03 danielw97

I've also tried this on ubuntu 22.04, and get a similar result. I'm training on ljspeech to give myself a good baseline and get familiar with the training process, my command is like this: python3 -m piper_train
--dataset-dir ~/piper/ljspeech
--accelerator 'gpu'
--devices 1
--batch-size 32
--validation-split 0.05
--num-test-examples 5
--max_epochs 1000
--checkpoint-epochs 1
--precision 32
--quality medium

danielw97 avatar Mar 31 '25 17:03 danielw97