lucidrains / naturalspeech2-pytorch

Implementation of Natural Speech 2, Zero-shot Speech and Singing Synthesizer, in Pytorch
MIT License
1.26k stars 100 forks source link

Torch, CUDA version #12

Open PS-AI opened 1 year ago

PS-AI commented 1 year ago

Hi,

Thanks for your work.

I am trying to run the training as mentioned in the Readme file. I am getting this error: RuntimeError: GET was unable to find an engine to execute this computation

I am using a Tesla V100 GPU.

Could you clarify what is the Torch. CUDA version you have used for your training.