我使用Tensorflow训练具有3个GPU的模型。但是,我发现它只使用1.代码没有任何GPU配置,因为我认为Tensorflow会自动检测GPU并自动将计算分配到3个GPU。它似乎不像我预期的那样工作。
从附加的屏幕截图中,在训练程序启动时检测到3个GPU,但温度监视器仅显示1个处于活动状态。
如何在培训期间同时使用3个GPU?
答案 0 :(得分:0)
您必须手动配置代码才能使用多个GPU进行培训。 因为必须为每个GPU操作定义上下文,您必须定义三个GPU的上下文,并分配或分配由相应GPU手动执行的任务。 请参考以下链接,它可能有助于您实现您的目标。 https://www.tensorflow.org/tutorials/using_gpu(适用于一般GPU配置)
https://www.tensorflow.org/tutorials/deep_cnn(这是关于多GPU的CNN培训的一个例子)