gpt4 book ai didi

cuda - 需要在多个 GPU 上分配数据的指导

转载 作者:行者123 更新时间:2023-11-30 09:57:59 24 4
gpt4 key购买 nike

我目前正在开发适用于 GPU 集群的机器学习工具包。我在多个 GPU 上测试了逻辑回归分类器。

我使用的是主从方法,其中主 CPU 创建多个 POSIX 线程,并且矩阵在 GPU 之间划分。

但是我遇到的问题是如何存储无法存储在单台机器上的大型矩阵。是否有任何库或方法可以在节点之间共享数据?

最佳答案

我不确定你的矩阵有多大,但你应该检查几周前发布的 CUDA 4.0。主要功能之一是跨多个 CUDA 设备/GPU 共享内存

关于cuda - 需要在多个 GPU 上分配数据的指导,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/6250178/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com