是否有解决方案限制 AMD OpenCL 平台使用的 GPU 数量?对于 NVIDIA 平台,只需设置环境变量即可CUDA_VISIBLE_DEVICES
限制 OpenCL 可用的 GPU 集。
编辑:我知道,我可以使用更少的设备创建一个上下文。但是,我正在寻找控制设备数量的方法for来自“外部”的 OpenCL 平台。
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 http://developer.amd.com/wordpress/media/2013/07/AMD_Accelerated_Parallel_Processing_OpenCL_Programming_Guide-rev-2.7.pdf
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)