如何解决Torch 并行化多个 GPU 的任务
我有两张 GPU 卡,并且我有目前在一张卡上运行的手电筒模型。我试着
使用这两张卡评估大型任务集。不幸的是,DataParallel
不适用于此类模型 (torchvision.models.detection.fasterrcnn_resnet50_fpn
)。
伪代码是
with GPUProcessPool() as p:
p.starmap(torch_task_function,iterable=[index,task for index,task_path in enumerate(paths)])
其中 GPUProcessPool()
将包含带有 GPU0
参数的过程和 GPU1
参数的过程。将这些任务拆分到两个 GPU 的最佳方法是什么?
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。