零、参考来源

利用CUDA查看多张显卡可用显存和总显存大小;
Modern CMake Adding the CUDA Language;

一、源文件写法

#include <cuda.h>
#include <cuda_runtime.h>
#include <device_launch_parameters.h>

#include <iostream>

int main()
{
    size_t avail(0);//可用显存
    size_t total(0);//总显存
    int nCudaNums = 0;

    cudaGetDeviceCount(&nCudaNums);//获取显卡数量

    
    for (int i = 0 ;i<nCudaNums;i++)
    {
        cudaSetDevice(i);

        cudaMemGetInfo(&avail,&total);
        //可以在这儿输出  
    }
    return 0；
}

二、CMake文件

cmake_minimum_required(VERSION 3.1)

project(cuda_test LANGUAGES CXX)

#cuda支持检测
enable_language(CUDA)
if(NOT CUDA_FOUND)
    message("cuda not found")
endif(NOT CUDA_FOUND)
include(CheckLanguage)
check_language(CUDA)

if(NOT DEFINED CMAKE_CUDA_STANDARD)
    set(CMAKE_CUDA_STANDARD 11)
    set(CMAKE_CUDA_STANDARD_REQUIRED ON)
endif()

add_library(cuda_test
    SHARED
    main.cu
)

target_link_libraries(CoreAlgo
    cudart
)