写一段代码,判断gpu是否可用
时间: 2023-03-27 11:04:03 浏览: 198
Tensorflow 测试一段能运行在 GPU 的代码
以下是判断GPU是否可用的代码示例:
#include <stdio.h>
#include <cuda_runtime.h>
int main() {
int deviceCount;
cudaError_t error = cudaGetDeviceCount(&deviceCount);
if (error != cudaSuccess) {
printf("CUDA error: %s\n", cudaGetErrorString(error));
return 1;
}
if (deviceCount == ) {
printf("There is no available GPU device.\n");
return 1;
}
printf("There are %d available GPU devices.\n", deviceCount);
return ;
}
阅读全文