PFLD icon indicating copy to clipboard operation
PFLD copied to clipboard

how to train use multi-gpu with your code

Open lzg188 opened this issue 5 years ago • 1 comments

大神您好,问一下,这tf-code能用多gpu训练吗?怎么设置,tf新手

lzg188 avatar Nov 01 '19 08:11 lzg188

可以参考一下tensorflow官网教程,将构建网络的代码写在with tf.device('gpu:#') 下

sljlp avatar Dec 05 '19 16:12 sljlp