如何从活动的CUDA设备获取属性?

gre*_*man 5 cuda nvidia

从CUDA设备(!)获取属性的众所周知的代码是枚举所有设备,并从中获取属性.然后我看到这样一块,激活给定的设备.

我有反向问题 - 假设已经选择了设备,我想获得它的属性(活动的),而不是系统中存在的所有设备.

我希望我以正确的方式写这个,因为我是CUDA的新手.

tal*_*ies 9

只需调用cudaGetDevice()以获取活动上下文的设备编号,然后调用cudaGetDeviceProperties以获取该设备的属性.在代码中,类似于:

int device;
cudaGetDevice(&device);

struct cudaDeviceProp props;
cudaGetDeviceProperties(&props, device);
Run Code Online (Sandbox Code Playgroud)

[免责声明:用浏览器编写,从未编译或测试过.自担风险]