根据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]]
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)