Python 使用多个GPU'；s

我有几个变量，我正在尝试执行以下操作：

np.array([ yHat[idS == i] for i in outIds])

---

这里，

outid

、

id

和

yHat

是相对较大的numpy数组。有没有办法得到同样的结果，但更快？我正在尝试获取正确ID的

yHat

值。顺序很重要。我有多个GPU的，所以一个多GPU的解决方案也将是伟大的

Nvidia CUDA可能是相关的

temp=dict（zip（id，yHat））；数组（[temp[i]表示outid中的i]）

似乎要快得多！目前我正在使用这个。。。感谢您的帮助。Nvidia CUDA可能是相关的

temp=dict（zip（id，yHat））；数组（[temp[i]表示outid中的i]）

似乎要快得多！目前我正在使用这个。。。谢谢你的帮助。