MockingBird 现在支持双GPU训练吗？

现在支持双GPU训练吗？

Open demoxiong opened this issue 3 years ago • 5 comments

抱歉，还没来得及仔细读代码，先在这问一下。现在的代码是否支持双GPU训练？

Jan 13 '22 09:01 demoxiong

刚看了代码，好像只支持单GPU训练。

Jan 13 '22 10:01 demoxiong

。。。。其实pytorch是支持的，但是需要共享一些变量，没双卡也没来得及搞

Jan 14 '22 10:01 babysor

好像可以双卡，我试了batchsize=48，两个显卡各用了10G左右

Jan 16 '22 09:01 demoxiong

你好,问一下安装的cuda和对应的源都是支持 GPU训练可以使用别人提供的模型进行播放但要训练自己的模型出现CPU满跑 GPU 不进行运算显卡是2060 请问 GPU训练是需要改动那些文件么

Oct 09 '22 08:10 mochaYJ12138

解决了吗？你可以看一下#501，最近也在学习这个

Nov 08 '22 16:11 by517840374