最近在跑Diffsinger,但是发现它好像不支持多卡运算,不知道是不是我什么地方弄错了,如果有知道什么实现Diffsinger多卡运算或者有教程的朋友麻烦发我一下
如果想要实现多卡运算,需要改动模型的代码,将模型分布在多个卡上运行,
通过数据并行的方式进行计算。
在TensorFlow中,可以使用tf.distribute API来实现多卡运算。
具体实现方式请参考TensorFlow的官方文档:https://tensorflow.google.cn/guide/distributed_training