内容纲要

参考链接: 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方法取出来
最后修改日期: 2019年4月25日

作者

留言

撰写回覆或留言

发布留言必须填写的电子邮件地址不会公开。