deviceId для использования функции cublasXtDeviceSelect

Мне нужно использовать функцию

cublasXtDeviceSelect (дескриптор cublasXtHandle_t, int nbDevices, int deviceId[])

из кубласной библиотеки. У меня есть два графических процессора, но я хочу работать с одним из них. Мне интересно, как я могу получить идентификатор устройства? Я знаю, что это 0 или 1, но я не знаю, какое из них является идентификатором устройства, на котором выполняется мой код. Этот идентификатор основан на порядке, в котором cudaGetDeviceProperties показывает результат в следующем коде?

HANDLE_ERROR( cudaGetDeviceCount( &count ) );
for (int i=0; i< count; i++) {
    HANDLE_ERROR( cudaGetDeviceProperties( &prop, i ) );
    printf( " --- General Information for device %d ---\n", i ); 

0 ответов

Другие вопросы по тегам