ОШИБКА: pyopencl: создание контекста для конкретного устройства

Я хочу создать контекст для конкретного устройства на моей платформе. Но я получаю ошибку.

Код:

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 
    ctx = cl.Context(devices[0])
AttributeError: 'Device' object has no attribute '__iter__'

Программа компилируется и выполняется без ошибок и предупреждений, если я использую:

ctx = cl.create_some_context()

Но мне придется вручную выбирать тип устройства каждый раз, когда я запускаю программу, используя эту функцию. Я могу установить следующую переменную среды

PYOPENCL_CTX='0'

Используя это, я не смогу создавать контексты для различных доступных устройств на основе требований. По умолчанию будет установлено устройство 0 для всех контекстов, которые я создаю.

Может кто-нибудь, пожалуйста, помогите мне с этой проблемой.

Спасибо

Ответы на вопрос(1)

Ваш ответ на вопрос