cudaChooseDevice
时间: 2023-10-26 12:12:44 浏览: 15
`cudaChooseDevice` 是 NVIDIA CUDA 库中的一个函数,用于在多个可用的 GPU 设备中选择一个设备来使用。函数的原型为:
```c
cudaError_t cudaChooseDevice(int* device,
const cudaDeviceProp* prop);
```
其中,`device` 参数是一个指向整数的指针,表示选择的设备的索引号将存储在该指针指向的位置。`prop` 参数是一个指向 `cudaDeviceProp` 结构体的指针,表示要选择设备的属性参数。
`cudaChooseDevice` 函数将返回一个 `cudaError_t` 类型的错误码,表示函数执行的状态。如果函数执行成功,将返回 `cudaSuccess`。
使用 `cudaChooseDevice` 函数可以在多个 GPU 设备中选择一个设备来执行 CUDA 程序,从而实现并行计算。