guodong's blog

master@zhejiang university
   

python多进程使用tensorflow多GPU

参考链接: https://github.com/tensorflow/tensorflow/issues/8220

只需要把improt tensoflow 放进进程模块内,而不是外。

同时可以指定环境变量 os.envrn[‘CUDA_VISIBLE_DEVICES’] = ‘{}’.format(t ask_id % gup_num)

顺便复习一下多进程:


from multiprogress import Pool

p=Pool(task_num) # 指定进程数,创建进程池
result = [] # 用于存放每个进程的返回值
for i in range(task_num):
    result.append(p.apply_async(fun, (args)))
p.close()
p.join()    
print([i.get() for i in result]) # results里面存放的是map对象,需要用get方法取出来



上一篇:
下一篇:

头像

guodong

没有评论


你先离开吧:)



发表评论

电子邮件地址不会被公开。 必填项已用*标注