错误：pyopencl：为特定设备创建上下文答案

【问题标题】：ERROR: pyopencl: creating context for specific device错误：pyopencl：为特定设备创建上下文
【发布时间】：2012-10-21 07:29:04
【问题描述】：

我想在我的平台上为特定设备创建上下文。但是我遇到了一个错误。

代码：

import pyopencl as cl
platform = cl.get_platforms()
devices = platform[0].get_devices(cl.device_type.GPU)
ctx = cl.Context(devices[0])

我得到的错误：

Traceback (most recent call last):
  File "D:\Programming\Programs_OpenCL_Python\Matrix Multiplication\3\main3.py", line 16, in <module>
    ctx = cl.Context(devices[0])
AttributeError: 'Device' object has no attribute '__iter__'

如果我使用，程序编译和执行没有任何错误和警告：

ctx = cl.create_some_context()

但每次使用此功能执行程序时，我都必须手动选择设备类型。我可以设置如下环境变量

PYOPENCL_CTX='0'

使用它，我将无法根据要求为可用的不同设备创建上下文。对于我创建的所有上下文，它将默认设置为设备 0。

谁能帮我解决这个问题。

谢谢

【问题讨论】：

标签： python opencl pyopencl

【解决方案1】：

根据 PyOpenCL 文档，Context 采用设备列表，而不是特定设备。

如果您将上下文创建代码更改为：

platform = cl.get_platforms()
my_gpu_devices = platform[0].get_devices(device_type=cl.device_type.GPU)
ctx = cl.Context(devices=my_gpu_devices)

它应该工作。如果您真的想将选择限制为仅一种设备，您可以操作my_gpu_devices 列表，例如：

my_gpu_devices = [platform[0].get_devices(device_type=cl.device_type.GPU)[0]]

【讨论】：

非常感谢。你解决了我的问题。顺便说一句，这也有效：platform = cl.get_platforms(); my_gpu_devices = 平台[0].get_devices(cl.device_type.GPU); ctx = cl.Context([my_gpu_devices[0]])
Yash 的版本也适用于我。 K. Brafford，我认为您的最后一行代码有错误。实际上我必须将上下文创建更改为 ctx = l.Context(devices=[my_gpu_devices[0]])