10分钟
GPU 参数
gpu_platform_id: 一个整数,表示OpenCL platform ID。通常每个GPU供应商都会 公开一个OpenCL platform。 默认为-1(表示系统级的默认platform)gpu_device_id: 一个整数,表示设备ID。 默认为-1 。 在选定的platform上,每个GPU都有一个唯一的设备ID。-1表示选定platform上的默认设备。gpu_use_dp: 一个布尔值,默认值为False。 如果为True, 则在GPU上计算时使用双精度(否则使用单精度)
学员评价