gpt4 book ai didi

python - 让 Pycuda 在 2 个 gpus 上与 Mpi4py 一起工作

转载 作者:行者123 更新时间:2023-11-28 18:54:05 24 4
gpt4 key购买 nike

<分区>

我正在尝试在两个 GPU 上运行一个 pycuda 程序。我读过 Talonmies 的一篇很棒的文章,解释了如何使用线程库来完成它,该文章还提到这可以通过 mpi4py 实现。

当我用 pycuda 运行 mpi4py 时,程序给出错误: self.ctx = driver.Device(gpuid).max_context pycuda._driver.logicError: cuDeviceGet 失败:未初始化

也许这是因为我试图同时初始化两个 gpu 设备。有没有人有一个非常简短的例子来说明我们如何让 2 个 gpus 与 mpi4py 一起工作?

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