从CUDA设备(!)获取属性的众所周知的代码是枚举所有设备,并从中获取属性。然后我看到这样的片段,激活给定的设备。
我有反向问题 - 假设已经选择了设备,我想获得它的属性(活动的),而不是系统中存在的所有设备。
我希望我以正确的方式写这个,因为我是CUDA的新手。
答案 0 :(得分:9)
只需调用cudaGetDevice()
即可获取活动上下文的设备编号,然后调用cudaGetDeviceProperties
以获取该设备的属性。代码类似于:
int device;
cudaGetDevice(&device);
struct cudaDeviceProp props;
cudaGetDeviceProperties(&props, device);
[免责声明:用浏览器编写,从未编译或测试过。使用风险自负]