【发布时间】:2017-12-25 02:26:16
【问题描述】:
我写了一个在 GPU 上计算的函数,然后我得到了一些数据并将它们转换为 python numpy.ndarray 形式。有什么方法可以检查数据的存储设备(CPU或GPU)吗?
我使用minpy支持GPU并行化得到一个array.ndArray类型的数组,应该保存在GPU内存上,然后我使用asnumpy函数将其转换为numpy数组形式。现在我想知道是否有办法找到转换后的 numpy 数组数据在哪里。
【问题讨论】:
-
这是一个奇怪的问题,因为 CPU 和 GPU 是处理器单元 (PU),而不是存储设备 (SD)。您能否提供一些上下文,也许是一些代码,用于您想要实现的最小、可重现的示例?
-
@PauloScardine 也许他们的意思是 GPU 上的视频内存?
-
也许检查数据类型:
arr.dtype,其中arr是数据。 -
@juanpa.arrivillaga 你认为他想知道某个 Python 对象是存储在主板的 RAM 内存中还是存储在视频板的专用 RAM 内存中?这是一个很长的镜头,OP真的应该用一些上下文来更新这个问题。也许最重要的问题是:为什么 OP 关心这个?
-
@PauloScardine 我试图在我的解释中保持慈善。 OP 确实更新了,现在我认为这实际上可能是他们的意图。还不是特别清楚。