限制AMD OpenCL的GPU数量

Question

是否有解决方案来限制AMD OpenCL平台使用的GPU数量？对于NVIDIA平台,可以简单地设置环境变量CUDA_VISIBLE_DEVICES以限制OpenCL可用的GPU集.

编辑:我知道,我可以用一组减少的设备创建一个上下文.不过,我正在寻找方法来控制设备的数量为从"外部"的OpenCL平台.

Answer 1

AMD拥有适用于Windows和Linux的GPU_DEVICE_ORDINAL环境变量.这允许您指定要从OpenCL应用程序中看到的GPU的索引.例如:

jprice@nowai:~/benchmark$ python benchmark.py -clinfo

Platform 0: AMD Accelerated Parallel Processing
 -> Device 0: Tahiti
 -> Device 1: Tahiti
 -> Device 2: Intel(R) Core(TM) i5-3550 CPU @ 3.30GHz

jprice@nowai:~/benchmark$ export GPU_DEVICE_ORDINAL=0
jprice@nowai:~/benchmark$ python benchmark.py -clinfo

Platform 0: AMD Accelerated Parallel Processing
 -> Device 0: Tahiti
 -> Device 1: Intel(R) Core(TM) i5-3550 CPU @ 3.30GHz

更详细的描述可以在AMD APP OpenCL编程指南(当前在第2.4.3节"屏蔽可见设备"中)中找到:http: //developer.amd.com/wordpress/media/2013/07/AMD_Accelerated_Parallel_Processing_OpenCL_Programming_Guide-rev- 2.7.pdf