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
上计算时使用双精度(否则使用单精度)
学员评价