CUVID를위한 장치의 GPU 아키텍처를 얻는 방법?

Question

NVidia의 NvDec CUVID 기능을 사용하여 비디오 디코더를 구현하고 있습니다. (비참하게 부적절한) 매뉴얼의 2 장에 따르면, 디코딩 제한은 GPU 아키텍처에 의해 지정됩니다. 즉, 최대 h265 수평 해상도는 GP10x에서 8192, GP100에서 4096 이하이며 GM206 미만의 아키텍처에서는 지원되지 않습니다.

CUDA를 사용하여 이러한 아키텍처를 검색하는 방법은 무엇입니까? 계산 기능이나 그로부터 추측할까요? 그리고 내가 그것을 추측해야한다면, 아키텍쳐 대 컴퓨팅 능력의 테이블이 있습니까?

Answer 1

GPU의 코드 이름을 반환하는 함수는 없지만 NVIDIA는 사용자가 기본 하드웨어 비디오 디코더의 기능을 쿼리 할 수 ​​있도록 cuvidGetDecoderCaps() API를 제공합니다.

cuvidGetDecoderCaps()의 상세한 예는 에서 다운로드 할 수 있으며, nvenc official site에서 다운로드 할 수 있습니다. Samples/NvDecodeD3D11/NvDecodeD3D11.cpp의 일례 :

CUVIDEOFORMAT videoFormat = g_pVideoSource->format(); 
CUVIDDECODECAPS videoDecodeCaps = {}; 
videoDecodeCaps.eCodecType = videoFormat.codec; 
videoDecodeCaps.eChromaFormat = videoFormat.chroma_format; 
videoDecodeCaps.nBitDepthMinus8 = videoFormat.bit_depth_luma_minus8; 
if (cuvidGetDecoderCaps(&videoDecodeCaps) != CUDA_SUCCESS) 
{ 
    printf("cuvidGetDecoderCaps failed: %d\n", result); 
    return; 
} 
if (!videoDecodeCaps.bIsSupported) { 
    printf("Error: This video format isn't supported on the selected GPU."); 
    exit(1);; 
}