在多个GPU的情况下, 利用cudaChooseDevice
函数选择符合条件的Device(GPU).
这个API很有用, 允许开发者选择合适的GPU执行程序.
选择版本大于等于5.0的GPU.
book.h
见 CUDA编程实战 - 准备工作
#include "book.h"
int main(void)
{
cudaDeviceProp prop;
int dev;
HANDLE_ERROR(cudaGetDevice(&dev));
printf("ID of current CUDA device: %d\n", dev);
memset(&prop, 0, sizeof(cudaDeviceProp));
prop.major=5;
prop.minor=0;
HANDLE_ERROR(cudaChooseDevice(&dev, &prop));
printf("ID of CUDA device closest to revision 1.3: %d\n", dev);
HANDLE_ERROR(cudaSetDevice(dev));
return 0;
}