我有一个深度学习工作站,其中有4个GPU,每个GPU具有6 GB的内存。是否可以使Docker容器将4个GPU视为一个但具有24 GB的GPU?

谢谢。

最佳答案

我以前从未使用过docker,但在使用Multiple GPU的CUDA上却做了很多工作。由于多个GPU在物理上是分开的,因此使用多个GPU需要在代码级别进行大量内存同步。

我不认为Docker可以虚拟合并所有GPU内存,因为这会使GPU端的计算非常复杂。使用多个GPU需要自定义内核才能相互同步。

我所提到的最好的比喻是:“您能否让两台裸机计算机合并RAM并像运行单台计算机一样运行Microsoft Word?”。

关于docker - 多个GPU是否可以作为一个具有更多内存的GPU工作?,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/60143380/

10-16 18:15