是否可以在pytorch中安装不同的显卡并使用多GPU?

Is it possible to install different graphic cards and use multi-GPU in pytorch?

我有一个问题。

pytorch是否可以安装不同的显卡并使用多GPU? 还有其他问题吗?

例> 3070(1ea)+3080(1ea)的组合,pytorch的数据并行功能可以吗?

提前感谢您的回复。

我相信在技术层面上是可行的,但它不是最优的。在不同 GPU 之间分配负载的代码假设它们是相同的。拥有不同的 GPU 基本上意味着您将根据所拥有的 GPU 的 最低 性能受益。
例如,如果您有一张内存为 1GB 的卡和另一张内存为 10GB 的卡,您将只能处理适合 1GB 卡的批次,而第二张卡有 9GB 未使用。