#include <stdio.h>
#include <cuda_runtime.h>
int main()
{
int iDeviceCount = 0; // 定义一个变量来存储系统中的 CUDA 设备数量
// 调用 cudaGetDeviceCount 获取设备数量,检查是否成功获取,或设备数量是否为0
if (cudaGetDeviceCount(&iDeviceCount) != cudaSuccess || iDeviceCount == 0)
{
// 如果检测不到 CUDA 设备或有错误发生,打印错误信息并返回1
printf("No CUDA devices found: %s\n", cudaGetErrorString(cudaGetLastError()));
return 1;
}
else
{
printf("CUDA device count: %d\n", iDeviceCount);
}
int iDev = 0; // 定义一个变量 iDev,用于指定要使用的设备ID,这里设为0
// 尝试设置当前线程的活动设备为 iDev(第0号设备)
if (cudaSetDevice(iDev) != cudaSuccess)
{
// 如果设置设备失败,打印错误信息
printf("CUDA set device error: %s\n", cudaGetErrorString(cudaGetLastError()));
}
else
{
// 如果设置设备成功,打印当前设置的设备ID
printf("CUDA set device: %d\n", iDev);
}
return 0;
}
cuda编程学习:获取gpu的数量和设置使用哪个gpu
最新推荐文章于 2024-09-11 14:42:29 发布