网站保定网站建设多少钱_洛阳seo网络推广_b站黄页推广软件_网络营销概述ppt

## main.pyfrom mpi4py import MPI# 下面例子是8GPU上跑24个进程来加速数据处理，因为一个model不能全站一个gpu显存，为了减少浪费，所以开最大进程来沾满
def main(args):comm = MPI.COMM_WORLDrank = comm.Get_rank()proc_num = comm.Get_size()gpu_rank = rank % 8# model = build_model.to(gpu_rank)datasets = get_datasets()start = (len(datasets) // proc_num + 1) * rankend = (len(datasets) // proc_num + 1) * (rank + 1)cur_sub_datasets = datasets[start:end]for data in datasets:out = model(data)save(out)if __name__ == '__main__':'''Total dataset: 68750.'''args = parse_args()if not os.path.exists(args.output_root_path):os.mkdir(args.output_root_path)main(args)

• 执行mpi4py支持的脚本

# -n 8 * x
mpirun -n 24 python main.py

Reference

• Semantic-Segment-Anything