是否使用其他GPU恢复训练Caffe？

Question

请原谅可能是一个疯狂的问题。

我的设置：两台机器，一台装有GTX1080，一台装有RX Vega64。在GTX1080上重新训练/微调bvlc_googlenet模型。

如果我为Vega 64构建Caffe，那么我可以从GTX1080机器上拍摄快照并重新开始在Vega 64上进行培训吗？从训练将以正常方式继续进行的意义上说，这项工作是否可行？

如果我将GTX1080快照移动到AWS中的Volta V100，该怎么办？这行得通吗？

我知道Caffe会在某种程度上抽象出硬件，但是我不知道它能做到多好。我需要GTX 1080来做其他事情...

谢谢！

Answer 1

据我所知，这应该没有问题。重量文件和训练快照只是数字的斑点，您应该能够在其他硬件（例如CPU / GPU），具有不同操作系统的其他计算机上或在32至64位进程之间恢复。